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 的等效函数.

备注

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