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