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)
输出 :
[2 4 6 8]