查找两个NumPy数组的并集可以使用union1d()
函数。该函数从两个输入数组中获取唯一的元素,然后返回一个排序的新数组。
以下是具体步骤:
- 导入NumPy库
import numpy as np
- 定义两个数组
arr1 = np.array([1,2,3,4])
arr2 = np.array([3,4,5,6])
- 使用
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]
在这个示例中,arr1
和arr2
数组的并集包含来自两个数组的所有唯一元素。可以看到,输出结果是一个排序的数组,仅包含唯一元素。