numpy求两个数组交集
时间: 2023-09-04 16:10:56 浏览: 286
python获得两个数组交集、并集、差集的方法
5星 · 资源好评率100%
可以使用numpy库的intersect1d方法来求两个数组的交集。
例如,假设有两个数组a和b:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5])
b = np.array([3, 4, 5, 6, 7])
```
可以使用intersect1d方法来求a和b的交集:
```python
c = np.intersect1d(a, b)
print(c)
```
输出结果为:
```
[3 4 5]
```
其中,c即为a和b的交集数组。
阅读全文