Python numpy.in1d()函数

numpy.in1d()函数测试一个一维数组中的每个元素是否也存在于第二个数组中，并返回一个与arr1相同长度的布尔数组，当arr1的一个元素在arr2中时为真，否则为假。

语法: numpy.in1d(arr1, arr2, assume_unique = False, invert = False)

参数 :
arr1 : [array_like] 输入阵列。
arr2 : [array_like] 用来测试arr1的每个值的值。
assume_unique : [bool, optional] 如果为真，则假设输入的数组都是唯一的，这可以加快计算速度。默认为假。
invert : [bool, optional] 如果为真，返回的数组中的值将被倒置。默认为假。

返回 : [ndarray, bool] arr1[in1d] 的值在 arr2 中。

代码#1：

# Python program explaining
# numpy.in1d() function
    
# importing numpy as geek 
import numpy as geek 
   
arr1 = geek.array([0, 1, 2, 3, 0, 4, 5])
arr2 = [0, 2, 5]
   
gfg = geek.in1d(arr1, arr2)
   
print (gfg)

输出 :

[ True False True False True False True]

代码#2：

# Python program explaining
# numpy.in1d() function
    
# importing numpy as geek 
import numpy as geek 
   
arr1 = geek.array([0, 1, 2, 3, 0, 4, 5])
arr2 = [0, 2, 5]
   
gfg = geek.in1d(arr1, arr2, invert = True)
   
print (gfg)

输出 :

[False True False True False True False]