[英]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
。 实际上,您实际上不需要为此编写任何循环,因为map
和set
类型的&
运算符将为您处理所有这一切:
intersection_set = set(map(tuple, list1)) & set(map(tuple, list2))
如果您需要的结果是一个list
的list
S,你当然可以,转换set
的tuple
的背部成list
的list
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.