[Numpy-discussion] Indexing by label in 3rd dimension
ben.root at ou.edu
Fri Sep 16 11:34:42 EDT 2011
On Friday, September 16, 2011, Benjamin Root <ben.root at ou.edu> wrote:
> On Friday, September 16, 2011, Benjamin Landenberger <
benjamin.landenberger at imtek.uni-freiburg.de> wrote:
>> Hello list!
>> I have an array *mask* of shape (a, b) and another array *intensities*
>> of shape (N, a, b), where the values in *mask* range from 0 to N-1. It
>> is somehow similar to label arrays in scipy.ndimage.
>> Now I want to pick those entries from the first dimension of
>> *intensities* which are given by *mask*. The returned array shall again
>> wave shape (a, b).
>> Can this be done with fancy indexing?
>> Thank you,
> Try intensities[mask, xrange(a), xrange(b)].
> Untested, but it would work for 1d, haven't tried for 2d.
> Ben Root
Crap, that wouldn't work... Sorry.
-------------- next part --------------
An HTML attachment was scrubbed...
More information about the NumPy-Discussion