使用Python中（n-1）-D数组中的索引从n-D数组中切片/樱桃挑选元素[重复]

Question

这是一个例子

>>>a = np.array([[1,2],[3,5]])
>>>a
array([[1, 2],
       [3, 5]])
>>>b = np.array([0, 1])

我正在寻找一种方法来从数组 a 的第 0 行和第 1 行中选择第 0 和第 1 个元素（由数组 b 表示），如下所示：

>>>np.array([a[b[0]], a[b[1]]]) = np.array([1, 5])

Answer 1

按照所谓的高级索引，这样就可以了

a[[0,1],[0,1]]