numpy.setdiff1d

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

查找两个数组的集合差集。

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

参数:

ar1array_like

输入数组。

ar2array_like

输入比较数组。

assume_uniquebool

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

返回值:

setdiff1dndarray

ar1 中不在 ar2 中的值的 1D 数组。当 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])