更快的列表理解
Faster list comprehension
我有两个以元组形式包含元素的列表。
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]
我有两个以元组形式包含元素的列表。
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]