Python numpy setdiff1d用法及代码示例

本文简要介绍 python 语言中 numpy.setdiff1d 的用法。

用法:

numpy.setdiff1d(ar1, ar2, assume_unique=False)

求两个数组的集差。

返回中的唯一值ar1那些不在ar2.

参数：

ar1： array_like

输入数组。

ar2： array_like

输入比较数组。

assume_unique： bool

如果为 True，则假定输入数组都是唯一的，这样可以加快计算速度。默认为假。

返回：

setdiff1d： ndarray

ar1 中不存在于 ar2 中的一维值数组。结果在assume_unique=False 时排序，否则仅在输入排序时排序。

例子：

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