设计求两个集合的交集的代码
时间: 2024-04-29 22:25:34 浏览: 69
N个集合求交集C语言代码
5星 · 资源好评率100%
假设集合A和集合B分别为两个列表,可以使用Python中的set()函数将列表转换为集合,然后使用&运算符求两个集合的交集,最后将交集转换为列表。
示例代码如下:
```python
A = [1, 2, 3, 4, 5]
B = [3, 4, 5, 6, 7]
# 将列表转换为集合
set_A = set(A)
set_B = set(B)
# 求交集
intersection = list(set_A & set_B)
print(intersection) # 输出:[3, 4, 5]
```
另外,如果不想使用set()函数,也可以使用列表推导式来实现求交集的功能,示例代码如下:
```python
A = [1, 2, 3, 4, 5]
B = [3, 4, 5, 6, 7]
# 求交集
intersection = [x for x in A if x in B]
print(intersection) # 输出:[3, 4, 5]
```
这里使用了列表推导式,遍历集合A中的每个元素,如果该元素同时也在集合B中,就将其添加到交集中。
阅读全文