pythonmultidimensional-arraynumpy

Indexing numpy array with another numpy array

Suppose I have

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

and 

b = array([1,1])

I'd like to use b in index a, that is to do a[b] and get 4 instead of [[3, 4], [3, 4]]

I can probably do 

a[tuple(b)]

Is there a better way of doing it?

Thanks

Solution

  • According the NumPy tutorial, the correct way to do it is:

    a[tuple(b)]