numpy.ma.in1d

ma.in1d(ar1, ar2, assume_unique=False, invert=False)

测试数组的每个元素是否也存在于第二个数组中.

输出始终是一个掩码数组.有关更多详细信息,请参见 numpy.in1d .

我们建议在新代码中使用 isin 代替 in1d .

参见

isin

此函数的版本,保留 ar1 的形状.

numpy.in1d

ndarray 的等效函数.

示例

Try it in your browser!

>>> import numpy as np
>>> ar1 = np.ma.array([0, 1, 2, 5, 0])
>>> ar2 = [0, 2]
>>> np.ma.in1d(ar1, ar2)
masked_array(data=[ True, False,  True, False,  True],
             mask=False,
       fill_value=True)

Go BackOpen In Tab