Chris, Note that >>> where(condition) actually returns a tuple (one item for each dimension of a) and each element is an array with the index of when your condition occurs. Therefore if you want to extract the index itself, you need to write idx = where(a==1)[0][0] for the first element, and where(a==1)[0] for the array of all indices.
Best, Jonathan On Wed, Oct 6, 2010 at 8:53 AM, Thomas, Brian (GE Energy) < [email protected]> wrote: > Chris, > > You can use where() command to find index of a particular number in the > array. > > For e.g. to find index of number 1 in array a, you can say > idx = where(a==1) > > Regards, > Brian > > -----Original Message----- > From: [email protected] > [mailto:[email protected]] On Behalf Of Chris Withers > Sent: Wednesday, October 06, 2010 5:27 AM > To: [email protected] > Cc: [email protected] > Subject: [Numpy-discussion] index of a value in an array > > Hi All, > > Given an array such as: > > array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) > > How can I find the index of a particular number in the array? > > (ie: if it was a list, I'd do [1,2,3,4].index(3)) > > cheers, > > Chris > _______________________________________________ > NumPy-Discussion mailing list > [email protected] > http://mail.scipy.org/mailman/listinfo/numpy-discussion > _______________________________________________ > NumPy-Discussion mailing list > [email protected] > http://mail.scipy.org/mailman/listinfo/numpy-discussion > -- Jonathan Rocher, Enthought, Inc. [email protected] 1-512-536-1057 http://www.enthought.com
_______________________________________________ NumPy-Discussion mailing list [email protected] http://mail.scipy.org/mailman/listinfo/numpy-discussion
