来自索引数组的python numpy数组

Question

我有一个3d数组a = np.arange(108).reshape(6, 6, 3) 。 我想获取数组的某些索引，如i = np.array([[0, 1], [1, 3], [2, 1]])这样结果为[[3, 4, 5], [27, 28, 29], [39, 40, 41]] 。 我需要一种有效的方法来执行此操作，因为我的实际数组大得多。

Answer 1

从i提取第一维和第二维索引 ，然后使用高级索引 ：

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

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