查找两个数组中相同的值np.intersect1d()/也可用于求两个数组的交集
numpy.intersect1d(ar1, ar2, assume_unique=False,
return_indices=False)
例子:查找两个数组中相同的值
data1 = np.random.randint(0,10,10)
data2 = np.random.randint(0,10,10)
print (data1)
print (data2)
print (np.intersect1d(data1,data2))
结果
[7 1 0 3 9 3 4 5 4 4]
[8 2 6 1 4 1 0 7 9 5]
[0 1 4 5 7 9]