[Numpy-discussion] Find index of repeated numbers in array

Nils Wagner nwagner@iam.uni-stuttgart...
Wed Dec 10 10:47:13 CST 2008


On Wed, 10 Dec 2008 11:38:11 -0500
  Ross Williamson <rw247@astro.columbia.edu> wrote:
> Hi Everyone
> 
> I think I'm missing something really obvious but what I 
>would like to  
> do is extract the indexes from an array where a number 
>matches - For  
> example
> 
> data = [0,1,2,960,5,6,960,7]
> 
> I would like to know, for example the indices which 
>match 960 - i.e.  
> it would return 3 and 6
> 
> I could do this with a loop but I was wondering if there 
>was a built  
> in numpy function to do this?
> 
> BTW if anyone is interested I'm converting some idl code 
>to numpy and  
> trying to mmic the IDL function where
> 
> Cheers
> 
> Ross
> _______________________________________________
> Numpy-discussion mailing list
> Numpy-discussion@scipy.org
>  
>>> data
array([  0,   1,   2, 960,   5,   6, 960,   7])
>>> where(data==960)
(array([3, 6]),)

Nils



More information about the Numpy-discussion mailing list