更快的列表理解

问题描述

我有两个元素作为元组的列表。

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]