Re: [Numpy-discussion] Fwd: Getting subarray

[Steve Lianoglou]

Hi,

On Apr 23, 2007, at 11:30 AM, Tommy Grav 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)]

respectively will get you what you're after.


-steve


___
Numpy-discussion mailing list
Numpy-discussion@scipy.org
http://projects.scipy.org/mailman/listinfo/numpy-discussion

Re: [Numpy-discussion] Fwd: Getting subarray

[Steve Lianoglou]

 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[b == 1]

and

a[b == 0]

work too, btw.

-steve
___
Numpy-discussion mailing list
Numpy-discussion@scipy.org
http://projects.scipy.org/mailman/listinfo/numpy-discussion