How to return indices of values between two numbers in numpy array

Question

I would like to return the indices of all the values in a python numpy array that are between two values. Here is my code:

inEllipseIndFar = np.argwhere(excessPathLen * 2 < ePL < excessPathLen * 3)

But it returns an error:

inEllipseIndFar  = np.argwhere((excessPathLen * 2 < ePL < excessPathLen * 3).all())
ValueError: The truth value of an array with more than one element is ambiguous. Use 
a.any() or a.all()

I'd like to know if there is a way of doing this without iterating through the array. Thanks!

Hooked · Accepted Answer · 2014-11-06 18:43:29Z

28

Since > < = return masked arrays, you can multiply them together to get the effect you are looking for (essentially the logical AND):

>>> import numpy as np
>>> A = 2*np.arange(10)
array([ 0,  2,  4,  6,  8, 10, 12, 14, 16, 18])

>>> idx = (A>2)*(A<8)
>>> np.where(idx)
array([2, 3])

answered Nov 6, 2014 at 18:43

Hooked

88.9k46 gold badges197 silver badges271 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

JoshAdel · Accepted Answer · 2014-11-06 18:47:20Z

14

You can combine multiple boolean expressions by using parentheses and the correct operation:

In [1]: import numpy as np

In [2]: A = 2*np.arange(10)

In [3]: np.where((A > 2) & (A < 8))
Out[3]: (array([2, 3]),)

You can also set the result of np.where to a variable to extract the values:

In [4]: idx = np.where((A > 2) & (A < 8))

In [5]: A[idx]
Out[5]: array([4, 6])

answered Nov 6, 2014 at 18:47

JoshAdel

69.1k27 gold badges146 silver badges146 bronze badges

3 Comments

Hooked Over a year ago

Out of curiosity, is there any difference between multiplication and logical conjunction for True/False arrays?

JoshAdel Over a year ago

@Hooked - Not really, but I find the compound logical statements to be more readable

s3dev Over a year ago

Old comments, but for future readers: I ran timing on both (logical and multiplication) and the difference appears negligible.

Collectives™ on Stack Overflow

How to return indices of values between two numbers in numpy array

2 Answers 2

Comments

3 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

3 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related