python - NumPy 链式比较与两个谓词

Question

在 NumPy 中，我可以像这样生成一个 bool 数组:

>>> arr = np.array([1, 2, 1, 2, 3, 6, 9])
>>> arr > 2
array([False, False, False, False,  True,  True,  True], dtype=bool)

我们如何将比较链接在一起？例如:

>>> 6 > arr > 2
array([False, False, False, False,  True,  False,  False], dtype=bool)

尝试这样做会导致错误消息

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

Answer 1

据我所知，最接近的是使用 & , | , 和 ^ :

>>> arr = np.array([1, 2, 1, 2, 3, 6, 9])
>>> (2 < arr) & (arr < 6)
array([False, False, False, False,  True, False, False], dtype=bool)
>>> (2 < arr) | (arr < 6)
array([ True,  True,  True,  True,  True,  True,  True], dtype=bool)
>>> (2 < arr) ^ (arr < 6)
array([ True,  True,  True,  True, False,  True,  True], dtype=bool)

我不认为你能得到 a < b < c - 风格的链接工作。