1. 导入numpy包
import numpy as np
2.造数据
a = np.array([2,3,3,1])
b = np.array([1,3,5,7])
3. 求交集
numpy.intersect1d(ar1, ar2, assume_unique=False, return_indices=False)
用途： Find the intersection of two arrays
返回： Return the sorted, unique values that are in both of the input arrays.返回的值是排序后的唯一的值。
交集：就是两个元素共同持有的相同的元素。
- 返回交集
np.intersect1d(a,b)
array([1, 3])
- 返回索引和交集
np.intersect1d(a,b, return_indices=True)
(array([1, 3]), array([3, 1], dtype=int64), array([0, 1], dtype=int64))
- 返回重复元素的交集
np.intersect1d(a, b, assume_unique=True)
array([1, 3, 3])
- 返回重复元素的索引和交集
np.intersect1d(a, b, assume_unique=True, return_indices=True)
(array([1, 3, 3]),
array([3, 1, 2], dtype=int64),
array([ 0, -2, 1], dtype=int64))
