Python numpy.setdiff1d()用法及代码示例

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

用法： numpy.setdiff1d(arr1, arr2, assume_unique = False)

参数：
arr1:[数组]输入数组。
arr2:[数组]输入比较数组。
assume_unique:[bool]如果为True，则假定输入数组都是唯一的，这可以加快计算速度。默认值为False。

Return :[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]

相关用法

注：本文由纯净天空筛选整理自sanjoy_62大神的英文原创作品 numpy.setdiff1d() function in Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。