使用numpy屏蔽数组中的元素

Question

我试图屏蔽数组中的一些元素，这样数学运算就不会应用于这些元素。

我运行了这个代码

import numpy as np
inp = np.random.randn(5, 5)
c = np.random.randn(5,5)
mask = inp > 0

inp[mask] += c
print(inp)

但是我收到这个错误

ValueError：操作数无法与形状(25，(一起广播(5,5((25，(

Answer 1

inp += mask.astype(int) * c
# -- or simplified to:
inp += mask * c