Thanks ;-) It's pretty good, as I can apply it to any condition.
Cheers, Nivolas. -----Original Message----- From: [EMAIL PROTECTED] [mailto:[EMAIL PROTECTED] On Behalf Of Francesc Alted Sent: Tuesday, November 18, 2008 11:40 AM To: Discussion of Numerical Python Subject: Re: [Numpy-discussion] Getting indices from numpy array withcondition A Tuesday 18 November 2008, Nicolas ROUX escrigué: > Hi, > > Maybe this is not so clever, but I can't find it in the doc. > I need to get all indices/index of all occurrences of a value in a > numpy array > > > As example: > > a = numpy.array([1,2,3],[4,5,6],[7,8,9]) > I need to get the indice/index of all array elements where a[a>3] > > Any fast/easy way to write this ? Perhaps the nonzero() method is what you are looking for: In [9]: (a > 3).nonzero() Out[9]: (array([1, 1, 1, 2, 2, 2]), array([0, 1, 2, 0, 1, 2])) Cheers, -- Francesc Alted _______________________________________________ Numpy-discussion mailing list [email protected] http://projects.scipy.org/mailman/listinfo/numpy-discussion _______________________________________________ Numpy-discussion mailing list [email protected] http://projects.scipy.org/mailman/listinfo/numpy-discussion
