如何返回numpy数组中两个数字之间的值索引

Question

我想返回两个值之间的python numpy数组中所有值的索引。这里是我的代码：

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

但它返回一个错误：

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() 

我想知道是否有这样做没有通过数组进行迭代的方式。谢谢！

Answer 1

由于> < =回报掩盖阵列，可以它们相乘得到你正在寻找的效果（基本上是逻辑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]) 

Answer 2

您可以通过使用括号和正确的操作组合多个布尔表达式：

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]),) 

你也可以的np.where结果设置为一个变量来提取值：

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

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