numpy求两个数组的并集
时间: 2023-12-24 20:07:20 浏览: 116
python获得两个数组交集、并集、差集的方法
5星 · 资源好评率100%
可以使用numpy中的union1d函数来求两个数组的并集。
示例代码如下:
```python
import numpy as np
# 定义两个数组
a = np.array([1, 2, 3, 4])
b = np.array([3, 4, 5, 6])
# 求并集
c = np.union1d(a, b)
print(c) # 输出 [1 2 3 4 5 6]
```
其中,np.union1d函数的第一个参数为第一个数组,第二个参数为第二个数组,返回值为两个数组的并集。
阅读全文