numpy.ma.isin

ma.isin(element, test_elements, assume_unique=False, invert=False)

计算 element in test_elements，仅在 element 上广播。

输出始终是与 element 形状相同的掩码数组。有关更多详细信息，请参阅 numpy.isin。

另请参阅

in1d

此函数的扁平版本。

numpy.isin

ndarray 的等效函数。

备注

1.13.0 版中的新增功能。

示例

>>> import numpy as np
>>> element = np.ma.array([1, 2, 3, 4, 5, 6])
>>> test_elements = [0, 2]
>>> np.ma.isin(element, test_elements)
masked_array(data=[False,  True, False, False, False, False],
             mask=False,
       fill_value=True)