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

Ross Williamson rw247@astro.columbia....
Wed Dec 10 12:16:29 CST 2008


Thanks all

I was being dumb and forgot to initialize as array()

Cheers

Ross
On Dec 10, 2008, at 11:47 AM, Sturla Molden wrote:

> On 12/10/2008 5:38 PM, Ross Williamson 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
>
>>>> import numpy
>>>> a = numpy.array([0,1,2,960,5,6,960,7])
>>>> a == 960
> array([False, False, False,  True, False, False,  True, False],  
> dtype=bool)
>>>> idx, = numpy.where(a == 960)
>>>> idx
> array([3, 6])
>>>> idx.tolist()
> [3, 6]
>
>
> Sturla Molden
> _______________________________________________
> Numpy-discussion mailing list
> Numpy-discussion@scipy.org
> http://projects.scipy.org/mailman/listinfo/numpy-discussion
>



More information about the Numpy-discussion mailing list