如何根据另一个元组列表的前两个元素删除目标列表中的元组

问题描述 #

现有两个元组列表list1和list2，需删除list1中所有与list2内任意元组前两个元素完全匹配的元组。示例如下：

list1 = [('google', 'data', '1'), ('google', 'data', '2'), ('google', 'data', '3'), ('google', 'data', '4'), ('google', 'WORLD', '1')]
list2 = [('google', 'data', '1'), ('google', 'HELLO', '2'), ('google', 'BLA', '3')]
# 预期结果：[('google', 'WORLD', '1')]

高效解决方案 #

核心思路是先把list2中需要排除的前两元素对提取为集合（集合查找效率远高于列表），再通过列表推导式过滤list1：

# 提取list2中所有元组的前两个元素，转为集合（去重+快速查找）
exclude_pairs = {item[:2] for item in list2}
# 过滤list1，保留前两元素不在排除集合中的项
list1 = [item for item in list1 if item[:2] not in exclude_pairs]

print(list1)  # 输出符合预期的结果

方案说明 #

• 用集合存储需要排除的元素对，是因为集合的成员检查时间复杂度为O(1)，如果list2元素较多，相比每次遍历list2检查（O(n*m)复杂度），能大幅提升效率。
• 列表推导式是Python中过滤列表最简洁高效的方式，代码可读性也很强。

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