Python中的 numpy.setdiff1d()函数

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

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

Parameters :
arr1 : [array_like] Input array.
arr2 : [array_like] Input comparison array.
assume_unique : [bool] If True, the input arrays are both assumed to be unique, which can speed up the calculation. Default is False.

Return : [ndarray] 1D array of values in arr1 that are not in arr2. The result is sorted when assume_unique = False, but otherwise only sorted if the input is sorted.

代码#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)

输出 ：

[2 4 6 8]