Python numpy.setdiff1d函数方法的使用

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中setdiff1d方法的使用。

numpy.setdiff1d

numpy.setdiff1d(ar1, ar2, assume_unique=False) [source]

查找两个数组的差值。

返回ar1中不在ar2中的唯一值。

参数：

ar1 ：array_like

输入数组。

ar2 ：array_like

输入比较数组。

assume_unique ：bool

如果为True，则假定输入数组都是唯一的，

这可以加快计算速度。默认值为False。

返回值：

setdiff1d ：ndarray

ar1中不在ar2中的值的一维数组。

当承担前提为false时对结果进行排序，

否则仅在对输入进行排序时才进行排序。

例子

>>> a = np.array([1, 2, 3, 2, 4, 1])
>>> b = np.array([3, 4, 5, 6])
>>> np.setdiff1d(a, b)
array([1, 2])