Python下拾遗系列： python下numpy包实现交集获取共同的元素

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))