给定另一个numpy数组，将numpy.ndarray的某些元素替换为零

Question

在以下情况下，我需要了解最有效的方法。有一个形状为11k * 11k的numpy.ndarray，对于给定形状为11k的二进制numpy数组，我需要将其中某些行的所有元素强制为零。一个玩具示例可以描述如下：

输入：

"Principal": {
  "AWS": [
    "312226949769"
  ]
}

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

输出：

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

Answer 1

使用此-

y = np.multiply(x.T,ref).T

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