寻找两个NumPy数组之间的共同值
在NumPy中,我们可以借助intersect1d()找到两个数组的共同值。它将接受两个数组的参数,并将返回一个数组,其中所有的共同元素都将出现。
语法: numpy.intersect1d(array1,array2)
参数:两个数组。
返回:一个数组,其中所有的共同元素都会出现。
示例 1:
import numpy as np
ar1 = np.array([0, 1, 2, 3, 4])
ar2 = [1, 3, 4]
# Common values between two arrays
print(np.intersect1d(ar1, ar2))
输出:
[1,3,4]
示例 2:
import numpy as np
ar1 = np.array([12, 14, 15, 16, 17])
ar2 = [2, 4, 5, 6, 7, 8, 9, 12]
# Common values between two arrays
print(np.intersect1d(ar1, ar2))
输出:
[12]