切片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))]