numpy.union1d¶
-
numpy.
union1d
(ar1, ar2)[source]¶ 查找两个数组的并集。
返回两个输入数组中任一个中唯一的已排序数值组。
参数: ar1,ar2:array_like
输入数组。如果他们还没有1D,它们是扁平的。
返回: union1d:ndarray
输入数组的唯一的已排序联合。
也可以看看
numpy.lib.arraysetops
- 具有用于对数组执行设置操作的多个其他功能的模块。
例子
>>> np.union1d([-1, 0, 1], [-2, 0, 2]) array([-2, -1, 0, 1, 2])
要找到两个以上的数组的并集,请使用functools.reduce:
>>> from functools import reduce >>> reduce(np.union1d, ([1, 3, 4, 3], [3, 1, 2, 1], [6, 3, 4, 2])) array([1, 2, 3, 4, 6])