Python numpy.union1d函数方法的使用

手机查看 2025-05-14

numpy.union1d() 是 NumPy 提供的一个集合操作函数，用于返回两个数组的并集（去重后合并），并且结果是排序后的一维数组。numpy.union1d 是一个简单且实用的 NumPy 函数，用于计算两个一维数组的集合并集。它能够自动处理重复元素并返回排序后的唯一元素。这在数据处理、集合操作等场景中非常有用。本文主要介绍一下NumPy中union1d方法的使用。

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)

文档：numpy.union1d