切片numpy数组与其他数组

Question

我试图提取由x包含的一组索引指定的一个numpy数组y的子集，同时仍然留下y一些索引。 举个具体的例子。 设y具有形状(10,10,10,3)而x具有形状(7,7,3) 。 x的最后一个维度对应于y的前三个维度的索引信息。 也就是说，我想要一个有效的切片操作，结果如下：

for i in x.shape[0]:
    for j in x.shape[1]:
        z[i,j,:] = y[x[i,j,0],x[i,j,1],x[i,j,2],:]

理想情况下，无论x的维数是多少，答案都会有效。

一般来说， y将是N+1维，具有形状(...,N) ，而x将是Q+1维的形状(...,N) ，并且结果将具有与x 。

其动机是从矢量场中提取矢量子集。

Answer 1

这应该运作得相当好：

y[x[..., 0], x[..., 1], x[..., 2]]

一般来说：

y[tuple(np.rollaxis(x, -1))]