Python numpy.union1d函数方法的使用

NumPy（Numerical Python的缩写）是一个开源的Python科学计算库。使用NumPy，就可以很自然地使用数组和矩阵。NumPy包含很多实用的数学函数，涵盖线性代数运算、傅里叶变换和随机数生成等功能。本文主要介绍一下NumPy中union1d方法的使用。

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])