Thanks ;-) It's pretty good, as I can apply it to any condition. Cheers, Nivolas. -----Original Message----- From: numpy-discussion-bounces@scipy.org [mailto:numpy-discussion-bounces@scipy.org] 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 Numpy-discussion@scipy.org http://projects.scipy.org/mailman/listinfo/numpy-discussion