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

假设我有

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

和

b = array([1,1]) 

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

我应该可以的

a[tuple(b)] 

有没有更好的方法？

谢谢

请您参考如下方法:

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

a[tuple(b)]