numpy.ma.masked_invalid#

ma.masked_invalid(a, copy=True)[source]#

对数组中出现无效值(NaN 或 inf)的位置进行掩码。

此函数是 masked_where 的快捷方式,其 condition = ~(np.isfinite(a))。任何预先存在的掩码都将保留。仅适用于 NaN 或 inf 有意义的数据类型(即浮点类型)的数组,但接受任何 array_like 对象。

另请参阅

masked_where

在满足条件的位置进行掩码。

示例

>>> import numpy as np
>>> import numpy.ma as ma
>>> a = np.arange(5, dtype=float)
>>> a[2] = np.nan
>>> a[3] = np.inf
>>> a
array([ 0.,  1., nan, inf,  4.])
>>> ma.masked_invalid(a)
masked_array(data=[0.0, 1.0, --, --, 4.0],
             mask=[False, False,  True,  True, False],
       fill_value=1e+20)