numpy.ma.putmask#

ma.putmask(a, mask, values)[source]#

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

这是 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]])