numpy.ma.make_mask#

ma.make_mask(m, copy=False, shrink=True, dtype=<class 'numpy.bool'>)[source]#

从数组创建布尔掩码。

m 作为布尔掩码返回,必要或请求时创建副本。该函数可以接受任何可转换为整数的序列,或 nomask。不要求内容必须是 0 和 1,值为 0 的被解释为 False,其他所有值都被解释为 True。

参数:
m类数组

潜在的掩码。

copy布尔值,可选

是返回 m 的副本 (True) 还是返回 m 本身 (False)。

shrink布尔值,可选

如果 m 的所有值均为 False,是否将其收缩为 nomask

dtype数据类型,可选

输出掩码的数据类型。默认情况下,输出掩码的数据类型为 MaskType (bool)。如果数据类型是灵活的,则每个字段都具有布尔数据类型。当 mnomask 时,此参数将被忽略,在这种情况下总是返回 nomask

返回:
resultndarray

一个从 m 派生出的布尔掩码。

示例

>>> import numpy as np
>>> import numpy.ma as ma
>>> m = [True, False, True, True]
>>> ma.make_mask(m)
array([ True, False,  True,  True])
>>> m = [1, 0, 1, 1]
>>> ma.make_mask(m)
array([ True, False,  True,  True])
>>> m = [1, 0, 2, -3]
>>> ma.make_mask(m)
array([ True, False,  True,  True])

shrink 参数的效果。

>>> m = np.zeros(4)
>>> m
array([0., 0., 0., 0.])
>>> ma.make_mask(m)
False
>>> ma.make_mask(m, shrink=False)
array([False, False, False, False])

使用灵活的 dtype

>>> m = [1, 0, 1, 1]
>>> n = [0, 1, 0, 0]
>>> arr = []
>>> for man, mouse in zip(m, n):
...     arr.append((man, mouse))
>>> arr
[(1, 0), (0, 1), (1, 0), (1, 0)]
>>> dtype = np.dtype({'names':['man', 'mouse'],
...                   'formats':[np.int64, np.int64]})
>>> arr = np.array(arr, dtype=dtype)
>>> arr
array([(1, 0), (0, 1), (1, 0), (1, 0)],
      dtype=[('man', '<i8'), ('mouse', '<i8')])
>>> ma.make_mask(arr, dtype=dtype)
array([(True, False), (False, True), (True, False), (True, False)],
      dtype=[('man', '|b1'), ('mouse', '|b1')])