python - 测试numpy数组中的每个元素是否位于两个值之间的简单方法？

Question

我想知道是否有一种语法上简单的方法来检查 numpy 数组中的每个元素是否位于两个数字之间。

换句话说，就像 numpy.array([1,2,3,4,5]) < 5将返回 array([True, True, True, True, False]) ，我想知道是否可以做类似的事情:

1 < numpy.array([1,2,3,4,5]) < 5

...获得...

array([False, True, True, True, False])

我知道我可以通过 bool 测试的逻辑链接来获得这一点，但我正在处理一些相当复杂的代码，并且我正在寻找一个语法上干净的解决方案。

有什么建议吗？

Answer 1

一种解决方案是:

import numpy as np
a = np.array([1, 2, 3, 4, 5])
(a > 1) & (a < 5)
# array([False,  True,  True,  True, False])