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

文档numpy.union1d

推荐阅读
cjavapy编程之路首页