numpy.
union1d
找到两个数组的并集。
返回两个输入数组中任意一个的值的唯一排序数组。
输入数组。如果它们还不是1d,则被压扁。
输入数组的唯一、已排序的联合。
参见
numpy.lib.arraysetops
具有许多其他函数的模块,用于对数组执行集合操作。
实例
>>> 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])