如何在Python中高效移除主列表中存在于其他指定列表的元素

当然有！你的核心需求是快速过滤主列表中存在于s1或s2的元素，原方案的痛点在于列表的in操作是O(n)时间复杂度，再加上remove操作需要移动列表元素的O(m)开销，当数据量较大时效率会很差。

最优思路：利用集合的O(1)查找特性 #

集合的成员查询基于哈希表实现，时间复杂度为O(1)，远快于列表的O(n)。我们可以先把s1和s2合并成一个排除集合，再通过简洁的方式快速过滤主列表：

s1 = [1,2,3,4,7]
s2 = [3,4,5,6,20]
mainlist = [6,7,8,9,10,11,12,13,14,15]

# 合并两个列表为排除集合（集合自动去重，不影响过滤逻辑）
exclude_set = set(s1) | set(s2)
# 列表推导式生成结果，无需显式遍历删除操作
resultList = [item for item in mainlist if item not in exclude_set]

执行后resultList的结果就是[8,9,10,11,12,13,14,15]，完全符合需求。

函数式风格替代写法：使用filter函数 #

如果你偏好函数式编程风格，也可以用filter配合匿名函数实现，效果和列表推导式一致：

resultList = list(filter(lambda item: item not in exclude_set, mainlist))

时间复杂度对比 #

• 原方案：O(len(mainlist) * (len(s1)+len(s2)) + len(mainlist)²)（in查询的线性开销+remove移动元素的二次开销）
• 新方案：O(len(s1)+len(s2)+len(mainlist))（集合构建的线性时间+一次遍历过滤的线性时间）

当列表元素数量超过几百个时，新方案的效率提升会非常明显。

内容的提问来源于stack exchange，提问作者Suhail Pappu