numpy.ma.in1d#

ma.in1d(ar1, ar2, assume_unique=False, invert=False)[源代码]#

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

输出始终是一个掩码数组。

我们建议在新代码中使用 isin 而不是 in1d。

另请参阅

isin: 此函数用于保留 ar1 的形状的版本。

示例

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