numpy.setdiff1d

麻木的。setdiff1d ( ar1 , ar2 , Should_unique = False )

求两个数组的集合差。

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

参数：

ar1类似数组

输入数组。

ar2类似数组

输入比较数组。

假定唯一布尔值

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

返回：

setdiff1d ndarray

ar1中不存在于ar2中的值的一维数组。当假设_unique=False时，结果将被排序，但否则仅当输入已排序时才会排序。

也可以看看

numpy.lib.arraysetops

具有许多其他函数的模块，用于对数组执行集合操作。

例子

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