How can I find the indices of an array with float values in python?

Peter Otten __peter__ at web.de
Thu Jan 10 12:25:34 EST 2019


Madhavan Bomidi wrote:

> I have an array (numpy.ndarray) with shape (1500L,) as below:
> 
> x = array([  3.00000000e+01,   6.00000000e+01,   9.00000000e+01, ...,
>          4.49400000e+04,   4.49700000e+04,   4.50000000e+04])
> 
> Now, I wanted to determine the indices of the x values between 0.0 and
> 15.0. While this is simple in MATLAB or IDL by using find or where
> functions, I was unable to find a best way to find the indices of all
> elements between 0.0 and 15.0 in the x array. Can you please suggest me a
> solution for the same?

Like this?

>>> a = numpy.array([-1, 10, 100, 5, 1000.])
>>> numpy.where((a < 15) & (a > 0))[0]
array([1, 3])





More information about the Python-list mailing list