[Numpy-discussion] Getting subarray

Tommy Grav tgrav@cfa.harvard....
Mon Apr 23 10:24:09 CDT 2007


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


More information about the Numpy-discussion mailing list