使用numpy屏蔽阵列

Question

我有一个数组，我想屏蔽它，这样我就可以保持它的形状，即，不删除屏蔽的元素。

例如在此代码中

input = torch.randn(2, 5)

mask = input > 0
input = input[mask]
input = input *1000000000000

print(input)

打印输入是对未屏蔽元素进行上述数学运算的结果，并返回一个没有屏蔽元素的1D数组。

Answer 1

执行input=input[mask]时，将覆盖原始数组。如果省略该步骤，则可以在适当位置修改屏蔽值，但保持非屏蔽值不变

i = np.random.randn(2, 5)

print(i)
[[ 0.48857855  0.97799014  2.29587523 -2.37257331  1.28193921]
 [ 0.62932172  1.37433223 -1.2427145   0.31424802  1.34534568]]

mask = i> 0
i[mask] *= 1000000000000

print(i)

[[ 4.88578545e+11  9.77990142e+11  2.29587523e+12 -2.37257331e+00 1.28193921e+12]
 [ 6.29321720e+11  1.37433223e+12 -1.24271450e+00  3.14248021e+11 1.34534568e+12]]