numpy.setdiff1d#
- numpy.setdiff1d(ar1, ar2, assume_unique=False)[源代码]#
查找两个数组的集合差集。
返回 ar1 中存在但不在 ar2 中的唯一值。
- 参数:
- ar1array_like
输入数组。
- ar2array_like
输入比较数组。
- assume_unique布尔值
如果为 True，则假设输入数组都是唯一的，这可以加快计算速度。默认为 False。
- 返回:
- setdiff1dndarray
ar1 中存在但不在 ar2 中的值的一维数组。当 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])