Python numpy.intersect1d()函数
numpy.intersect1d()函数找到两个数组的交点,并返回两个输入数组中的排序后的唯一值。
语法: numpy.intersect1d(arr1, arr2, assume_unique = False, return_indices = False)
参数 :
arr1, arr2 : [array_like] 输入数组。
assume_unique : [bool] 如果为真,则假设输入数组都是唯一的,这可以加快计算速度。默认为假。
return_indices : [bool] 如果为真,则返回两个数组的交点所对应的索引。如果有多个值,则使用第一个实例。默认为假。
返回 : [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]