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