python - 如何将 numpy 中的值范围设置为 nan？

Question

我的数据集如下所示:

>>> difference
array([[ -1,   0,   4],
       [-20,   2,  -1],
       [  2, -20,   0]])

我想用 nan 替换从 +2 到 -2 的值。

结果数组应该如下所示。

>>> difference 
array([[ nan,  nan,   4.],
       [-20.,  nan,  nan],
       [ nan, -20.,  nan]])

Answer 1

使用 np.abs(difference)<=2要获得范围的真值，一种方法是使用 np.where , 像这样 -

np.where(np.abs(difference)<=2,np.nan,difference)

sample 运行-

In [5]: difference
Out[5]: 
array([[ -1,   0,   4],
       [-20,   2,  -1],
       [  2, -20,   0]])

In [6]: np.where(np.abs(difference)<=2,np.nan,difference)
Out[6]: 
array([[ nan,  nan,   4.],
       [-20.,  nan,  nan],
       [ nan, -20.,  nan]])

为了完整起见，获得这种掩码的更明确的方法是使用 (difference <=2) & (difference >=-2)而不是 np.abs(difference)<=2 .