NumPy之求数组集合运算

一、交集

intersect1d的作用是用来求两个数组的交集。

1.1、参数介绍

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

ar1,ar2：输入数组，如果不是一维数组将会被展平成一维数组。
assume_unique:如果为 True，则假定输入数组都是唯一的，这可以加快计算速度。如果 True 但 ar1 或 ar2 不是唯一的，则可能导致不正确的结果和越界索引。默认为假
return_indices：如果为 True，则返回对应于两个数组交集的索引。如果有多个值，则只返回第一个值。

返回值

intersect1d：排序后的两个输入数组中的交集元素
comm1：交集元素在第一个数组中的序号
comm2：交集元素在第二个数组中的序号

1.2、使用方法

In [1]: