我有一个大小为(1080,1920,3)的图像存储在一个numpy数组中，我试图使用大小为(6,2)的索引数组对其进行索引。我的代码在下面并运行，但输出尺寸为(2,6,3)，这不是我所期望的。
A = inPix[np.arange(6), 2darray]
上述方法来自答案here。任何人都可以帮助我理解什么是错的以及如何获得(6,3)输出数组？
答案 0 :(得分：0)
我发现如果你将索引分成列，它就可以了。
A = inPix[2darray[:,0],2darray[:,1]]