[Numpy-discussion] Fwd: Getting subarray
Gael Varoquaux
gael.varoquaux at normalesup.org
Mon Apr 23 11:49:51 EDT 2007
On Mon, Apr 23, 2007 at 11:44:08AM -0400, Steve Lianoglou wrote:
> > I have two arrays:
> > a = numpy.array([0,1,2,3,4,5,6,7,8,9])
> > b = numpy.array([0,0,1,1,2,2,0,1,2,3])
> > I would like to get the part of a that corresponds
> > to where b is equal to i.
> > For example:
> > i = 0 => ([0,1,6])
> > i = 1 => ([2,3,7])
> a[numpy.where(b == 0)]
> and
> a[numpy.where(b == 1)]
How about "a[ b==0 ]" and "a[ b==1 ]" ?
Gaël
More information about the NumPy-Discussion
mailing list