NumPy参考 >例行程序 >Set routines > numpy.setdiff1d
numpy.
setdiff1d
(ar1,ar2,假定_unique = False )[源代码] ¶找到两个数组的集合差。
返回的唯一值AR1不在AR2。
输入数组。
输入比较数组。
如果为True,则假定输入数组都是唯一的,这可以加快计算速度。默认值为False。
ar1中不在ar2中的值的一维数组。当前提条件为authentic_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])