是否可以用布尔表达式索引NumPy数组？

Question

假设我有以下数组：

a = np.array([0,1,0],
             [1,0,0],
             [0,0,1])

是否可以做这样的事情：

a[==0] = -1

设置所有零为-1？ 我知道您可以通过以下方式达到相同的效果：

b = a == 0
a[b] = -1

但是我只是想知道是否有一种方法可以跳过创建布尔数组的方法。

Answer 1

您的语法几乎是正确的。 这似乎可行：

a[a == 0] = -1

Answer 2

使用np.where

np.where(a==0, -1, a)