python数据分析numpy基础之intersect1d求数组交集

1 python数据分析numpy基础之intersect1d求数组交集

python的numpy库的intersect1d(x,y)函数，计算x和y的公共元素，并返回去重后的有序结果。相当于求两个数组中的交集，并且进行去重和排序。

用法

numpy.intersect1d(ar1, ar2, assume_unique=False, return_indices=False)

描述

numpy.intersect1()查找两个数组的公共元素，并且返回去重后的有序结果。

入参

ar1，ar2：必选，列表、元组、数组；如果是多维则自动转一维数组；

assume_unique：可选，bool，默认False，若为True则可以加快计算素材，此时ar1和ar2需为元素唯一的数组，否则结果不会去重；

return_indices：可选，bool，默认False，若为True则返回公共元素在ar1和ar2的索引，如果存在多个相同的值，则返回第一个值查询的索引。

1.1 入参ar1和ar2

numpy. intersect1d()的入参ar1和ar2，为必选入参，可以为数组、列表、元组。如果是多维数组，将会转换为一维数组后，进行处理。

>>> import numpy as np
# intersect1d()返回两个数组的公共元素，并且去重和排序
# 入参ar1/ar2为列表
>>> np.intersect1d([1,2,3,5,2],[6,5,2,1,5])
array([1, 2, 5])
# 入参ar1/ar2为元组
>>> np.intersect1d((1