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_uniquebool

如果为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])
推荐阅读
cjavapy编程之路首页