Re: [Numpy-discussion] Indexing a 2-d array with a 1-d mask

2011-02-09 Thread Alok Singhal
On Wed, Feb 9, 2011 at 1:24 AM, Friedrich Romstedt wrote: > 2011/2/8 Alok Singhal : >> In [6]: data2 = numpy.zeros((0, 5), 'd') >> In [7]: mask2 = numpy.zeros(0, 'bool') >> In [8]: data2[mask2] >> >> Traceback (most recent call last): >>

Re: [Numpy-discussion] Indexing a 2-d array with a 1-d mask

2011-02-09 Thread Friedrich Romstedt
Hi, 2011/2/8 Alok Singhal : > Hi, > > I have an NxM array, which I am indexing with a 1-d, length N boolean > array.  For example, with a 3x5 array: > > In [1]: import numpy > In [2]: data = numpy.arange(15) > In [3]: data.shape = 3, 5 > > Now, I want to select rows 0 and 2, so I can do: > > In [4

[Numpy-discussion] Indexing a 2-d array with a 1-d mask

2011-02-08 Thread Alok Singhal
Hi, I have an NxM array, which I am indexing with a 1-d, length N boolean array. For example, with a 3x5 array: In [1]: import numpy In [2]: data = numpy.arange(15) In [3]: data.shape = 3, 5 Now, I want to select rows 0 and 2, so I can do: In [4]: mask = numpy.array([True, False, True]) In [5]