numpy.union1d#
- numpy.union1d(ar1, ar2)[source]#
查找两个数组的并集。
返回两个输入数组中存在的唯一排序的数组值。
- 参数:
- **ar1, ar2**array_like
输入数组。如果它们不是一维的,则会将其扁平化。
- 返回:
- **union1d**ndarray
输入数组的唯一排序并集。
示例
>>> import numpy as np >>> 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])