如何用另一字典的键替换字典值集中元素并避免重复？

问题需求 #

需要将字典my_dict中值为集合的元素，用dict_vals的键替换：若my_dict集合中的元素存在于dict_vals的任一值集合中，则替换为对应的dict_vals键，且替换后每个集合无重复元素。

给定输入：

my_dict = {0:{2,3}, 1:{0,4,5}, 2:{0,1,5,7}}
dict_vals = {'val1':{1,2,3},'val2':{7,8}}

期望输出：

my_dict = {0:{'val1'}, 1:{0,4,5}, 2:{0,'val1',5,'val2'}}

错误代码分析 #

你提供的代码存在多处核心问题，导致无变化甚至会报错：

• 逻辑完全搞反：你检查的是val in keys_replace（即元素是否是dict_vals的键），但实际需要判断的是元素是否在dict_vals的值集合里。比如my_dict里的元素是2、3、7这些，根本不在dict_vals的键（'val1'、'val2'）中，所以条件永远不触发，自然没有替换操作。
• 集合不支持索引操作：集合是无序且不可通过下标访问的，my_dict[key][i]这种写法会直接抛出TypeError，因为集合没有索引。
• 遍历集合用enumerate无意义：集合本身没有固定顺序，enumerate拿到的索引对集合来说毫无作用。

正确实现方式 #

先构建反向映射字典，把dict_vals中每个值集合里的元素映射到对应的键，这样后续替换时可以直接查表，效率更高：

my_dict = {0:{2,3}, 1:{0,4,5}, 2:{0,1,5,7}}
dict_vals = {'val1':{1,2,3},'val2':{7,8}}

# 构建反向映射：元素 -> 对应替换的键
reverse_map = {}
for replace_key, elem_set in dict_vals.items():
    for elem in elem_set:
        reverse_map[elem] = replace_key

# 遍历处理my_dict的每个集合
for key in my_dict:
    new_elements = set()
    for elem in my_dict[key]:
        # 存在映射则替换，不存在则保留原元素
        new_elements.add(reverse_map.get(elem, elem))
    my_dict[key] = new_elements

print(my_dict)
# 输出：{0: {'val1'}, 1: {0, 4, 5}, 2: {0, 5, 'val1', 'val2'}}

说明 #

1. 反向映射reverse_map会生成{1: 'val1', 2: 'val1', 3: 'val1', 7: 'val2', 8: 'val2'}，这样每个需要替换的元素都能直接找到对应的键。
2. 遍历my_dict的每个集合时，用reverse_map.get(elem, elem)实现“有映射则替换，无映射则保留”的逻辑。
3. 用新集合存储处理后的元素，最后替换原字典中的集合，避免在遍历集合时修改原集合导致的问题。

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