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