索引数组中的python numpy数组

我有一个3d数组

a=np.arange（108）。重塑（6,6,3）

。我想获取数组的某些索引，如

I=np.array（[[0,1]，[1,3]，[2,1]]）

所定义，这样结果就是

[[3,4,5]，[27,28,29]，[39,40,41]

。我需要一种有效的方法来实现这一点，因为我的实际数组要大得多。

从

I

中提取第一维度和第二维度索引，然后使用：

a[i[:,0], i[:,1], :]   # or a[i[:,0], i[:,1]]

#array([[ 3,  4,  5],
#       [27, 28, 29],
#       [39, 40, 41]])