[Numpy-discussion] yet another indexing question

Neal Becker ndbecker2 at gmail.com
Fri Oct 14 08:04:39 EDT 2011


suppose I have:

In [10]: u
Out[10]: 
array([[0, 1, 2, 3, 4],
       [5, 6, 7, 8, 9]])

And I have a vector v:
 v = np.array ((0,1,0,1,0))

I want to form an output vector which selects items from u where v is the index 
of the row of u to be selected.

In the above example, I want:

w = [0,6,2,8,4]

I can't seem to find a syntax that does this.

Now, more importantly, I need the result to be a reference to the original array 
(not a copy), because I'm going to use it on the LHS of an assignment.  Is this 
possible?




More information about the NumPy-Discussion mailing list