找到两个 NumPy 数组的并集

NumPy 是 Python 语言的一个扩展程序库，支持大量的维度数组与矩阵运算。在 NumPy 中可以执行各种数值计算，因此它在科学、工程、数据分析等领域得到了广泛的应用。

在 NumPy 中，可以使用函数 numpy.union1d() 来找到两个数组的并集。此函数将两个数组中的元素进行合并，并去除重复的元素。

示例

import numpy as np

arr1 = np.array([1, 2, 3])
arr2 = np.array([2, 3, 4])

union = np.union1d(arr1, arr2)

print(union)

输出：

[1 2 3 4]

代码解释：

• 首先导入 NumPy 库
• 定义两个数组 arr1 和 arr2，分别包含元素 [1, 2, 3] 和 [2, 3, 4]
• 使用 np.union1d() 函数获取 arr1 和 arr2 的并集，保存在变量 union 中
• 输出 union，结果为 [1, 2, 3, 4]

思考

• 如果两个数组中有相同元素，如何去重？
• 如果需要找到多个数组的并集，应该如何实现？
• 如果需要找到多个数组的交集或差集，应该如何实现？