带有2D数组的索引图像



我的大小(1080,1920,3)的图像存储在numpy阵列中,并试图使用大小(6,2)的索引阵列进行索引。我的代码在下面,它运行,但是输出尺寸为 (2,6,3),这不是我期望的。

A = inPix[np.arange(6), 2darray]

以上方法来自此处的答案。谁能帮助我了解什么问题以及如何获得(6,3)输出数组?

我发现,如果将索引分为列,则可以工作。

A = inPix[2darray[:,0],2darray[:,1]]

最新更新