如何解决更快的列表理解
lst1 = [(101,201),(102,203),(103,204)]
lst2 = [(1,101,(2,202),(3,102,(4,103,204),(5,205)]
如果第二个和第三个项目与第一个列表中的元组匹配,我需要输出第二个列表中元组中的第一项。我正在使用列表理解并获得预期结果,但我在两个列表中都有接近百万个元素,所以速度太慢了。
out_lst = [x[0] for x in lst2 if (x[1],x[2]) in lst1]
# Output
[1,3,4]
有没有更快的方法?
解决方法
您可以使用 set
,因为搜索 set
对象的时间是恒定的。
lst1 = [(101,201),(102,203),(103,204)]
lst2 = [(1,101,(2,202),(3,102,(4,103,204),(5,205)]
set1 = set(lst1)
res = [x[0] for x in lst2 if (x[1],x[2]) in set1]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。