NumPy 找到两个数组的并集

NumPy是一个流行的Python库，支持数值计算。它被广泛应用于科学计算、数据分析和机器学习中的数组和矩阵操作。在NumPy中，最常见的任务之一就是找到两个数组的并集。当两个数组合并时，会创建一个新的数组，其中包含来自两个数组的所有不同元素。在本文中，我们将探讨不同的方法来找到两个NumPy数组的并集。

安装和语法

通常使用Anaconda或Miniconda发行版安装NumPy。如果您尚未安装，可以使用pip安装。以下命令将安装最新版本的NumPy –

pip install numpy

NumPy库提供了一个名为 union1d() 的函数，用于找到两个数组的并集。

numpy.union1d(ar1, ar2)

步骤

要使用union1d()函数找到两个NumPy数组的并集，我们首先导入NumPy库。然后我们创建两个数组arr1和arr2。最后，我们使用 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]

构建两个数组， arr1 和 arr2 ，并向它们添加一些共享的组件，然后进一步确定这两个数组的并集 – 由 union1d() 方法确定。结果是一个包含来自两个数组的所有不同组件的新数组。

示例2

import numpy as np

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

union = np.union1d(arr1, arr2)
print(union)

输出

[1 2 3 4 5 6]

创建两个互不相同的独立数组 a1 和 a2，并且使用和之前的示例相同的方法来确定这两个数组的并集。结果将会输出两个集合的总数，因为它们之间没有重叠。

示例3

import numpy as np

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

union = np.union1d(arr1, arr2)
print(union)

输出

[1. 2. 3.]

创建一个空数组arr1和一个带有一些元素的数组arr2。然后使用union1d()函数找到这两个数组的并集。输出结果是一个新数组，其中包含来自两个数组的所有唯一元素。

示例4

import numpy as np

# create two arrays with some common elements
arr1 = np.array([1, 2, 3, 4, 5])
arr2 = np.array([4, 5, 6, 7, 8])

# find the union of the two arrays
union = np.union1d(arr1, arr2)

# print the union array
print(union)

输出

[1 2 3 4 5 6 7 8]

我们首先使用import numpy as np导入NumPy库。然后使用np.array()函数创建两个数组arr1和arr2，并且这两个数组有一些共同的元素。我们使用np.union1d()函数找到这两个数组的并集，并将结果赋给一个名为union的变量。最后，我们使用print()函数打印出并集数组。