是否可以使用布尔表达式索引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

你的语法几乎是正确的。这似乎有效：

for(var i = 1; i <= 5; i++) {

      for(var j = 1; j<= i; j++) {

        document.write("*");  

      }

      document.write("<br/>");
}

Answer 2

使用np.where

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