python - 如果我的行中的元素也为零，如何将整行设置为零？

Question

我有一个看起来像这样的矩阵

a = np.random.rand(2, 2, 2)

[[[1 0]
  [1 1]]

 [[1  1]
  [1 0]]]

我想使用 numpy 设置此矩阵中具有 0 到 0 的任何行。所以在这种情况下，它看起来像这样，

[[[0 0]
  [1 1]]

 [[1  1]
  [0 0]]]

有人可以给我提示吗？

Answer 1

与 0 进行比较，并沿最后一个轴 axis=-1(或第三个轴 axis=2)查找 ANY 匹配项 在这个特定的 3D 情况下)，然后使用这个掩码来分配零 -

a[(a==0).any(axis=-1)] = 0

示例运行 -

In [16]: a
Out[16]: 
array([[[1, 0],
        [1, 1]],

       [[1, 1],
        [1, 0]]])

In [17]: a[(a==0).any(-1)] = 0

In [18]: a
Out[18]: 
array([[[0, 0],
        [1, 1]],

       [[1, 1],
        [0, 0]]])