numpy.setdiff1d

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

查找两个数组的集合差。

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

参数：

ar1array_like

输入数组。

ar2array_like

输入比较数组。

assume_uniquebool

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

返回：

setdiff1dndarray

1D 数组，包含ar1中不在ar2中的值。assume_unique=False时结果已排序，否则只有在输入已排序时才排序。

示例

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