我的大小(1080,1920,3)
的图像存储在numpy阵列中,并试图使用大小(6,2)
的索引阵列进行索引。我的代码在下面,它运行,但是输出尺寸为 (2,6,3)
,这不是我期望的。
A = inPix[np.arange(6), 2darray]
以上方法来自此处的答案。谁能帮助我了解什么问题以及如何获得(6,3)
输出数组?
我发现,如果将索引分为列,则可以工作。
A = inPix[2darray[:,0],2darray[:,1]]