numpy.union1d
numpy.union1d(ar1, ar2) [source]
找到两个数组的并集。
返回两个输入数组之一中唯一的,排序的值数组。
参数 : | ar1, ar2 :array_like 输入数组。 如果还不是一维的,则将它们展平。 |
返回值 : | union1d :ndarray 输入数组的唯一,排序的并集。 |
例子
>>> 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])