使用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]]