查找两个NumPy数组的并集

  • Post category:Python

要查找两个NumPy数组的并集,我们可以使用NumPy库中的union1d()函数,该函数返回两个输入数组的并集。

下面是查找数组并集的完整攻略:

  1. 导入NumPy库

在使用NumPy库的操作之前,需要先导入NumPy库,使用以下代码即可:

import numpy as np
  1. 创建两个NumPy数组

在执行查找数组并集操作之前,需要先创建两个NumPy数组,可以使用以下代码创建两个NumPy数组:

a = np.array([1, 2, 3, 4, 5])
b = np.array([4, 5, 6, 7, 8])

创建完成后,NumPy数组 ab 分别为 [1,2,3,4,5][4,5,6,7,8]

  1. 使用 union1d() 函数查找数组并集

NumPy库中提供的 union1d() 函数可以用于查找两个数组的并集,使用以下代码即可查找数组 ab 的并集:

c = np.union1d(a, b)

执行完成后,NumPy数组 c 的值应该为 [1,2,3,4,5,6,7,8],即为数组 a 和数组 b 的并集。

下面是两个查找NumPy数组并集的示例:

示例一:

import numpy as np

a = np.array([1, 2, 3, 4, 5])
b = np.array([4, 5, 6, 7, 8])

c = np.union1d(a, b)

print(c)

输出结果为:

[1 2 3 4 5 6 7 8]

示例二:

import numpy as np

a = np.array([1, 1, 2, 2, 3, 3])
b = np.array([2, 2, 3, 3, 4, 4])

c = np.union1d(a, b)

print(c)

输出结果为:

[1 2 3 4]

在示例二中,数组 ab 中有重复的元素,但是在查找并集的过程中不会出现重复元素,结果数组 c 的值为 [1,2,3,4]