Re: [Numpy-discussion] Getting subarray

2007-04-24 Thread Bill Baxter
Easy! a[b==i] --bb On 4/24/07, Tommy Grav <[EMAIL PROTECTED]> 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])

[Numpy-discussion] Getting subarray

2007-04-24 Thread Tommy Grav
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]) Cheers Tommy ___ Nump