[Numpy-discussion] how to find array indices at which a condition is satisfied?
Ernest Adrogué
eadrogue@gmx....
Thu Aug 20 20:14:02 CDT 2009
20/08/09 @ 18:00 (-0700), thus spake Dr. Phillip M. Feldman:
> I have a 1-D array and would like to generate a list of indices for which a
> given condition is satisfied. What is the cleanest way to do this?
you can do something like this:
numpy.arange(len(x))[x > 5]
it'll give you the indices of x where x is > 5.
Ernest
More information about the NumPy-Discussion
mailing list