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