在numpy中，如何比较轴中的所有值

Question

对于 numpy 数组，只有当轴上的所有元素都等于另一个数组时，如何更改值？例如...

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

我想将所有内容替换[1, 0, 1]为[1, 1, 1]...，这样array就变成了

array([[1, 1, 1],
       [0, 0, 1],
       [1, 1, 0],
       [0, 0, 0],
       [1, 1, 1]])

当我使用布尔数组时，它会检查每个单独的数字。我怎样才能同时比较整行呢？

Answer 1

尝试使用：

array[(array == [1, 0, 1]).all(axis=1)] = [1, 1, 1]