查找两个NumPy数组的并集

  • Post category:Python

查找两个NumPy数组的并集可以使用union1d()函数。该函数从两个输入数组中获取唯一的元素,然后返回一个排序的新数组。

以下是具体步骤:

  1. 导入NumPy库
import numpy as np
  1. 定义两个数组
arr1 = np.array([1,2,3,4])
arr2 = np.array([3,4,5,6])
  1. 使用union1d()函数
union_array = np.union1d(arr1, arr2)
print(union_array)

输出结果为:

[1 2 3 4 5 6]

以上是其中一种方法,下面我们再来看一个示例:

import numpy as np

arr1 = np.array([10, 20, 30, 40, 50])
arr2 = np.array([40, 50, 60, 70])

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

输出结果为:

[10 20 30 40 50 60 70]

在这个示例中,arr1arr2数组的并集包含来自两个数组的所有唯一元素。可以看到,输出结果是一个排序的数组,仅包含唯一元素。