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