On Friday, September 16, 2011, Benjamin Landenberger < [email protected]> 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, > > Ben >
Try intensities[mask, xrange(a), xrange(b)]. Untested, but it would work for 1d, haven't tried for 2d. Ben Root
_______________________________________________ NumPy-Discussion mailing list [email protected] http://mail.scipy.org/mailman/listinfo/numpy-discussion
