用另一个 numpy 数组索引 numpy 数组

Question

假设我有

a = array([[1, 2],
           [3, 4]])

和

b = array([1,1])

我想在索引a中使用b，也就是做a[b]得到4而不是[[3, 4], [3, 4]]

我应该可以的

a[tuple(b)]

有没有更好的方法？

谢谢

Answer 1

根据NumPy tutorial，正确的做法是：

a[tuple(b)]

Answer 2

假设您要访问a 的子向量，其中有n 个索引对存储在b 中，如下所示：

b = array([[0, 0],
       ...
       [1, 1]])

这可以按如下方式完成：

a[b[:,0], b[:,1]]

对于单对索引向量，这将更改为a[b[0],b[1]]，但我猜tuple 方法更易于阅读，因此更可取。