numpy.setdiff1d#
- numpy.setdiff1d(ar1, ar2, assume_unique=False)[source]#
查找两个数组的集合差集。
返回 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])