numpy.ma.putmask

ma.putmask(a, mask, values)

根据条件和输入值更改数组的元素。

这是 numpy.putmask 的掩码数组版本，有关详细信息，请参阅 numpy.putmask。

另请参阅

numpy.putmask

注释

使用掩码数组作为 values 不会将 ndarray 转换为 MaskedArray。

示例

>>> import numpy as np
>>> arr = [[1, 2], [3, 4]]
>>> mask = [[1, 0], [0, 0]]
>>> x = np.ma.array(arr, mask=mask)
>>> np.ma.putmask(x, x < 4, 10*x)
>>> x
masked_array(
  data=[[--, 20],
        [30, 4]],
  mask=[[ True, False],
        [False, False]],
  fill_value=999999)
>>> x.data
array([[10, 20],
       [30,  4]])