找到两个array的交集,用python写
时间: 2023-08-21 17:07:57 浏览: 123
python实现把两个二维array叠加成三维array示例
5星 · 资源好评率100%
你可以使用 Python 的 `numpy` 库中的 `intersect1d()` 函数来查找两个数组的交集。
以下是一个示例代码:
```python
import numpy as np
def get_intersection(a, b):
intersection = np.intersect1d(a, b)
return intersection.tolist()
```
这个函数接受两个数组 `a` 和 `b` 作为输入,返回它们的交集。如果交集非空,则返回一个包含交集元素的列表;否则返回一个空列表。
你可以使用以下代码测试这个函数:
```python
a = np.array([1, 2, 3, 4, 5])
b = np.array([3, 4, 5, 6, 7])
c = np.array([8, 9, 10])
print(get_intersection(a, b)) # [3, 4, 5]
print(get_intersection(a, c)) # []
```
输出结果为:
```
[3, 4, 5]
[]
```
需要注意的是,使用 `numpy` 库需要先安装并导入库。如果你只需要找到两个列表的交集而不需要其他 `numpy` 库中的函数,可以使用前面提到的 Python 内置函数来实现。
阅读全文