python - 如何基于具有/相同维度和形状的掩码数组更新 numpy 数组

Question

我有两个具有相同尺寸和形状的数组。此 nv_arr 数组具有空值 (-9999)。我创建了一个掩码数组 nv_mask 来掩码除 -9999 之外的所有值，以某种方式按位置“更新”此 out 数组，而无需创建新数组。

换句话说，数字0、3、6和2必须替换为-9999。我怎样才能做到这一点？

    >>> nv_arr
     ([[-9999,   123,   -9999],
       [-9999,   444,   444],
       [-9999,   323,   323]])

>>> nv_mask
[[-9999 -- -9999]
 [-9999 -- --]
 [-9999 -- --]]

 >>> out
      ([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

Answer 1

假设nv_mask是一个MaskedArray，您可以调用.mask属性来设置值:

>>> out[~nv_mask.mask] = -9999
>>> out 
array([[-9999,     1, -9999],
       [-9999,     4,     5],
       [-9999,     7,     8]])