Python：移除字典短键值列表中与长键重复的元素

安全阿岳

2026-5-14

你的嵌套循环思路能实现需求，但算不上最优——它会重复遍历所有键值对，做不少无用功，而且没有利用键的长度优先级来简化逻辑。我们可以用更Pythonic的方式来实现，既简洁高效，又符合Python的惯用写法。

更优的实现思路

核心思路是：

用集合替代列表做成员检查，把O(n)的查找成本降到O(1)；
通过字典推导式一次性生成结果，避免修改原字典带来的潜在问题；
精准定位需要对比的长键，只处理短键中存在于长键列表的冗余值。

完整代码实现

original_dict = {"key1": ["val1", "val2", "val3"], "key1_key2": ["val1", "val3", "val4"]}

# 先把每个键的元素转成集合，提升成员检查效率
key_element_sets = {k: set(vals) for k, vals in original_dict.items()}

# 用字典推导式生成去重后的结果
cleaned_dict = {
    key: [val for val in vals if val not in set().union(*[
        key_element_sets[longer_key]
        for longer_key in original_dict
        if len(longer_key) > len(key) and key in longer_key
    ])]
    for key, vals in original_dict.items()
}

print(cleaned_dict)
# 输出: {'key1': ['val2'], 'key1_key2': ['val1', 'val3', 'val4']}

代码解释

key_element_sets：将每个键对应的列表转成集合，这样后续检查元素是否存在的操作会快很多，尤其当列表元素较多时效果明显。
字典推导式：直接生成新的干净字典，比循环修改原字典更安全、更符合Python风格。
set().union(*[...]：把所有符合条件的长键（长度更长且包含当前短键作为子串）的元素集合合并成一个大集合，这样我们只需要一次检查就能过滤掉所有冗余值。如果当前键没有对应的长键，这个合并集合会是空集，自然不会过滤任何元素，完美符合需求。