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

numpy.intersect1d()函数查找两个数组的交集，并返回两个输入数组中都有序的，唯一的值。

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

参数：
arr1，arr2：[数组]输入数组。
assume_unique:[bool]如果为True，则假定输入数组都是唯一的，这可以加快计算速度。默认值为False。
return_indices:[bool]如果为True，则返回与两个数组的交集相对应的索引。如果有多个值，则使用值的第一个实例。默认值为False。

Return :[ndarray]常见和唯一元素的排序一维数组。

代码1：

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

输出：

[1 2 4]

代码2：

# Python program explaining 
# numpy.intersect1d() 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] 
    
gfg = geek.intersect1d(arr1, arr2) 
    
print (gfg)

输出：

[1 3 5 7 9]

相关用法

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