从大型Python列表中删除大量元素,同时保留元素的顺序



假设我有两个列表,每个列表有100000个元素,我想找到集合的差异,但保持元素的顺序。示例:

A = [10, 1, 30, 4, 5, 6]
B = [4, 1, 100]

期望输出:

AB = [10, 30, 5, 6]

set(A).difference(set(B))正在生成[10, 5, 6, 30](注意它没有保留A中的顺序(,列表理解需要一些时间(我认为这是因为in运算符(。

一旦得到设置的差异,就可以按顺序将其与原始列表进行比较,然后相应地保存结果

A = [10, 1, 30, 4, 5, 6]
B = [4, 1, 100]
C = set(A).difference(set(B))
result = [item for item in A if item in C]

输出

[10, 30, 5, 6]  

使用列表综合

A = [x for x in A if x not in B]

相关内容

最新更新