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