numpy.union1d
numpy.union1d(ar1, ar2) [source]
找到两个数组的并集。
返回两个输入数组之一中唯一的，排序的值数组。
参数 ： | ar1, ar2 ：array_like 输入数组。 如果还不是一维的，则将它们展平。 |
返回值 ： | union1d ：ndarray 输入数组的唯一，排序的并集。 |
例子
import numpy as np
a = np.array([1, 2, 3])
b = np.array([3, 4, 5])
result = np.union1d(a, b)
print(result)
要查找两个以上数组的并集，请使用functools.reduce：
import numpy as np
from functools import reduce
# 定义多个数组
arrays = [
[1, 3, 4, 3],
[3, 1, 2, 1],
[6, 3, 4, 2]
]
# 通过 reduce 和 union1d 依次合并为并集
result = reduce(np.union1d, arrays)
print(result)