numpy.setdiff1d

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

找到两个数组的集合差.

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

参数:

ar1array_like

输入数组.

ar2array_like

输入比较数组.

assume_uniquebool

如果为真,则假定输入数组都是唯一的,这可以加快计算速度.默认值为假.

返回:

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])