繁体   English   中英

Python中两个嵌套列表的交集

[英]Intersection of two nested lists in Python

嵌套列表有问题。 我想用python语言计算两个嵌套列表的交集的长度。 我的清单组成如下:

list1 = [[1,2], [2,3], [3,4]]
list2 = [[1,2], [6,7], [4,5]]
output_list = [[1,2]]

如何计算两个列表的交集?

我认为有两种合理的方法可以解决此问题。

如果顶层列表中没有太多项目,则只需检查其中一个子列表中的每个子列表是否存在于另一个列表中:

intersection = [inner_list for inner in list1 if inner_list in list2]

in运算符将测试是否相等,因此可以按预期找到具有相同内容的不同列表对象。 但是,由于列表成员资格测试必须遍历所有子列表,因此效率不是很高。 换句话说,其性能为O(len(list1)*len(list2)) 但是,如果您的清单很长,可能要花费比您想要的更多的时间。

渐近有效的替代方法是将内部列表转换为tuple s,并将顶级list s转换为set 实际上,您实际上不需要为此编写任何循环,因为mapset类型的&运算符将为您处理所有这一切:

intersection_set = set(map(tuple, list1)) & set(map(tuple, list2))

如果您需要的结果是一个listlist S,你当然可以,转换settuple的背部成listlist S:

intersection_list = list(map(list, intersection_set))

在python中使用集合呢?

>>> set1={(1,2),(2,3),(3,4)}
>>> set2={(1,2),(6,7),(4,5)}
>>> set1 & set2
set([(1, 2)])
>>> len(set1 & set2)
1
import json



list1 = [[1,2], [2,3], [3,4]]
list2 = [[1,2], [6,7], [4,5]]

list1_str = map(json.dumps, list1)
list2_str = map(json.dumps, list2)

output_set_str = set(list1_str) & set(list2_str)

output_list = map(json.loads, output_set_str)

print output_list

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM