本文介绍如何安全高效地为 python 中 defaultdict(set) 某个键对应集合的所有元素添加统一前缀，避免在遍历中直接修改集合导致的逻辑错误和重复加前缀问题。

本文介绍如何安全高效地为 python 中 defaultdict(set) 某个键对应集合的所有元素添加统一前缀，避免在遍历中直接修改集合导致的逻辑错误和重复加前缀问题。

在使用 defaultdict(set) 管理分组数据时，常需对特定键（如 'key2'）下的所有字符串值批量添加前缀（例如 "foo-"）。初学者易采用「边遍历边修改」的方式，如下所示：

for i in my_set["key2"]:
    i_new = "foo-" + i
    my_set["key2"].remove(i)  # ⚠️ 危险：修改正在迭代的集合
    my_set["key2"].add(i_new)

该写法存在两个严重问题：

• 运行时异常风险：Python 不允许在遍历集合的同时增删其元素（虽部分版本未立即报错，但行为未定义）；
• 逻辑错误：remove(i) 后若新元素 "foo-i" 又被后续迭代命中，将再次加前缀，导致 "foo-foo-i" 等嵌套前缀（正如提问中出现的 'foo-foo-val3'）。

✅ 正确做法是避免原地修改，转而构建全新集合。利用集合推导式或生成器表达式一次性生成带前缀的新集合，并直接赋值给对应键：

my_set["key2"] = {f"foo-{item}" for item in my_set["key2"]}

或等价写法（显式调用 set() 构造）：

my_set["key2"] = set(f"foo-{item}" for item in my_set["key2"])

该方案优势显著：

• ✅ 安全：无迭代与修改冲突；
• ✅ 简洁：单行完成转换，语义清晰；
• ✅ 高效：时间复杂度 O(n)，且避免了多次哈希查找与内存重分配；
• ✅ 可扩展：轻松适配其他变换（如后缀、大小写转换、正则替换等），只需修改表达式内部逻辑。

? 提示：若需对多个键批量处理，可封装为函数：

def add_prefix_to_set(d: defaultdict, key: str, prefix: str) -> None:
    if key in d:
        d[key] = {f"{prefix}{item}" for item in d[key]}

# 使用示例
add_prefix_to_set(my_set, "key2", "foo-")

总之，面对集合的批量变换操作，请始终遵循「不可变思维」——以新代旧，而非原地修修补补。这既是 Python 的惯用实践，也是写出健壮、可维护代码的关键原则。