Numpy：使用一维数组索引 3D 矩阵

Question

arr = np.arange(12).reshape((3, 2, 2))
indices = np.array([0, 1, 1])

expected_outcome = np.array([[0, 1], [6, 7], [10, 11]])

我正在尝试使用包含我想要获取的值的 y 索引的形状数组 (3) 来索引这个形状数组 (3,2,2)。 我试图让它与for in语句一起工作，但有没有一种优雅的方法可以用 numpy 做到这一点？

Answer 1

所以你想要arr[0,0,:], arr[1,1,:], arr[2,1,:] ？

怎么样

In [179]: arr[[0,1,2], [0,1,1]]
Out[179]: 
array([[ 0,  1],
       [ 6,  7],
       [10, 11]])