python - 如何将 numpy.where 与逻辑运算符一起使用

Question

我正在尝试查找数组中大于 a 但小于 b 的所有元素的索引。这可能只是我的语法有问题，但这不起作用：

numpy.where((my_array > a) and (my_array < b))

我应该如何解决这个问题？或者有更好的方法吗？

谢谢！

Answer 1

这里有两种方法：

In [1]: my_array = arange(10)

In [2]: where((my_array > 3) & (my_array < 7))
Out[2]: (array([4, 5, 6]),)

In [3]: where(logical_and(my_array > 3, my_array < 7))
Out[3]: (array([4, 5, 6]),)

对于第一个（替换and为&），请注意适当地添加括号： &具有比比较运算符更高的优先级。你也可以使用*，但我不推荐它：它很老套，不利于代码的可读性。

In [4]: where((my_array > 3) * (my_array < 7))
Out[4]: (array([4, 5, 6]),)