python - 列表搜索优化

Question

first = [(1, text, text, 1, 2, 3), 
         (1, text, text, 1, 0, 3), ... (6054, text, text, 2, 2, 3)]
second = (1, 2, 3, 4, 5 ... 5412)

有没有更快的方法:

data = [x for x in first if x[0] in second]

Answer 1

试试这个:

first = [(1, text, text, 1, 2, 3), 
         (1, text, text, 1, 0, 3), ... (1054, text, text, 2, 2, 3)]
second = (1, 2, 3, 4, 5 ... 5412)
second_set = set(second)
data = [x for x in first if x[0] in second_set]

假设第一个有 m 个元素，第二个有 n 个元素。

集合经过哈希处理，因此搜索它们的时间接近于 O(1)，整体效率为 O(m)。搜索第二个列表的时间复杂度为 O(n)，总效率为 O(m * n)。