Python numpy.setdiff1d()函数

numpy.setdiff1d()函数找到两个数组的集合之差，并返回arr1中不在arr2中的唯一值。

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

参数 :
arr1 : [array_like] 输入阵列。
arr2 : [array_like] 输入比较阵列。
assume_unique : [bool] 如果为真，则假设输入数组都是唯一的，这可以加快计算速度。默认为假。

返回： [ndarray] arr1中不在arr2中的值的一维数组。当assume_unique = False时，结果会被排序，否则只有当输入被排序时才会被排序。

代码#1：

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

输出 :

[4 5 6]

代码#2：

# Python program explaining
# numpy.setdiff1d() function
    
# importing numpy as geek 
import numpy as geek 
   
arr1 = [1, 2, 3, 4, 5, 6, 7, 8, 9]
arr2 = [1, 3, 5, 7, 9, 11, 13, 15]
   
gfg = geek.setdiff1d(arr1, arr2)
   
print (gfg)