索引中的 NumPy 布尔数组

Question

假设我们有两个数组

a = np.array([1,2,3,4]).reshape(2,2)
b = np.array([True, False, False, True]).reshape(2,2)

给予

a = [[1, 2],
     [3, 4]] 
b = [[True, False],
     [False, True]]

我们可以做 a[b] 来只得到 b 的真实值给我们 [1, 4]

我尝试执行 a[not b] 来获取那些错误但出错的内容。我知道我可以做到a[b == False]，但我想以 Python 的方式做到这一点。

有什么办法吗？

Answer 1

您可以使用np.logical_not 或~ operator：

>>> a[np.logical_not(b)]
>>> a[~b]