首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >Python -多个列表的交集一次取两个。

问Python -多个列表的交集一次取两个。
EN

Stack Overflow用户
提问于 2022-11-24 08:42:08
回答 2查看 45关注 0票数 -1

在python中,我能够得到多个列表的交集:

代码语言:javascript
复制
arr = [[1, 2, 3], [2, 3, 4], [3, 4, 5]]
result = set.intersection(*map(set, arr))

输出:

代码语言:javascript
复制
result = {3}

现在,我希望将结果作为所有3个嵌套列表的交集,每次取2:

代码语言:javascript
复制
result = {2, 3, 4}

因为2,3在第1和第2列表之间是常见的,3、4在第2和第3列表之间是常见的,而3在第1和第3列表之间是常见的。

有内置的功能吗?

EN

回答 2

Stack Overflow用户

发布于 2022-11-24 08:54:29

你可以采取以下所有对交叉口的联合;

代码语言:javascript
复制
import itertools as it
arr = [[1, 2, 3], [2, 3, 4], [3, 4, 5]]
res = set.union(*(set(i).intersection(set(j)) for i,j in it.combinations(arr,2)))
# output {2, 3, 4}

*编辑per @DanielHao的评论

票数 1
EN

Stack Overflow用户

发布于 2022-11-24 08:46:01

试试itertools.combinations

代码语言:javascript
复制
from itertools import combinations
arr = [[1, 2, 3], [2, 3, 4], [3, 4, 5]]
result = [set.intersection(*map(set, i)) for i in combinations(arr,2)] 
# print(list(combinations(arr,2)) to get the combinations
# [{2, 3}, {3}, {3, 4}]
票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/74557859

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档