以任意维度对具有numpy数组的数组进行索引numpy数组

Question

假设img是2维numpy数组。假设x和y是与img相同形状的整数值2维numpy数组。考虑：

newImg = img[x, y] 

newImg现在是相同的形状的IMG 2维阵列，其中newImg [I，J] == IMG [X [I，J]，Y [I，J]]对于所有的i和j。

我想将此过程概括为任意数量的维度。也就是说，让img是一个d维numpy数组，并把x [i]作为一个整数值的d维numpy数组，它的形状与img相同。我基本想要的是：

newImg = img[x[0], x[1], ..., x[d-1]] 

这显然是伪代码，并没有预期的工作。

我该怎么用NumPy做到这一点？

Answer 1

你试过只是

newImg = img[x] 

看起来这应该工作！

我假设x是满足下列条件

len(x) == img.ndim 
all(a.shape == img.shape for a in x) 

这似乎符合你所描述的整数数组列表或元组。