python - 更快的列表理解
问题描述
我有两个元素作为元组的列表。
lst1 = [(101, 201), (102, 203), (103, 204)]
lst2 = [(1, 101, 201), (2, 101, 202), (3, 102, 203), (4, 103, 204), (5, 103, 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, 201), (2, 101, 202), (3, 102, 203), (4, 103, 204), (5, 103, 205)]
set1 = set(lst1)
res = [x[0] for x in lst2 if (x[1], x[2]) in set1]
推荐阅读
- javascript - 在 COR 响应期间收到服务器 500 错误
- java - ConstraintLayout:约束是可交换的吗?
- android - 如果侦听器已存在,如何防止应用创建新侦听器
- javascript - jQuery 使用路径参数验证远程选项
- actionscript-3 - 如何阻止玩家在墙上行走
- java - 正则表达式在两个取消标记之间获取单词
- azure - Azure Active Directory 是否支持将单个用户绑定到多个配置文件,每个配置文件都属于自己的组?
- python - 版本升级 1.0.0 后散景数据表选择事件不起作用
- docker - 在 AWS Fargate 容器中存储数据库连接字符串
- java - useDelimiter() 中的重复值