[Numpy-discussion] Indexing by label in 3rd dimension

Benjamin Landenberger benjamin.landenberger@imtek.uni-freiburg...
Fri Sep 16 10:18:11 CDT 2011


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,

Ben



More information about the NumPy-Discussion mailing list