找到两个数组的交集,用python写
时间: 2023-08-21 09:07:57 浏览: 81
python获得两个数组交集、并集、差集的方法
5星 · 资源好评率100%
你可以使用 Python 内置的 `set()` 函数和 `&` 运算符来查找两个列表的交集。
以下是一个示例代码:
```python
def get_intersection(a, b):
set_a = set(a)
set_b = set(b)
intersection = set_a & set_b
return list(intersection)
```
这个函数接受两个列表 `a` 和 `b` 作为输入,返回它们的交集。如果交集非空,则返回一个包含交集元素的列表;否则返回一个空列表。
你可以使用以下代码测试这个函数:
```python
a = [1, 2, 3, 4, 5]
b = [3, 4, 5, 6, 7]
c = [8, 9, 10]
print(get_intersection(a, b)) # [3, 4, 5]
print(get_intersection(a, c)) # []
```
输出结果为:
```
[3, 4, 5]
[]
```
需要注意的是,使用 `set()` 函数会导致原始列表中的元素顺序被打乱。如果需要保留原始列表中的元素顺序,可以使用列表推导式或者 `filter()` 函数来实现。
阅读全文