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:
输出 :
代码#2:
输出 :