numpy.ma.isin

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

计算 element in test_elements，仅对 element 进行广播。

输出始终是与 element 形状相同的掩码数组。更多详情请参见 numpy.isin。

另请参见

in1d

此函数的扁平化版本。

numpy.isin

ndarrays 的等效函数。

示例

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