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

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

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

и

b = array([1,1])

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

Я, наверное, смогу

a[tuple(b)]

Есть ли способ лучше?

Спасибо

20
задан xster 1 April 2011 в 01:40
поделиться