Индексирование массива numpy другим массивом

Question

Предположим, у меня есть

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

и

b = array([1,1])

Я бы хотел использовать b в индексе a, то есть сделать a [b] и получить 4 вместо [[3, 4], [3, 4]]

Я, наверное, могу сделать

a[tuple(b)]

Есть ли лучший способ сделать это?

Спасибо

Answer 1

Согласно учебному пособию:

a[tuple(b)] 

- правильный способ сделать это:

http://www.scipy.org/Tentative_NumPy_Tutorial#head-3f4d28139e045a442f78c5218c379af64c2c8c9e

Answer 2

Предположим, вы хотите получить доступ к подвектору a с n индексными парами, хранящимися в b, например:

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

Это можно сделать следующим образом:

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

Для вектора индекса одной пары это меняется на a[b[0],b[1]], но я думаю, что подход tuple легче читать и, следовательно, предпочтительнее.