如何以更优、更Pythonic的方式反转字典中键与值的引用关系？

嘿，我来给你捋几个更清爽、更贴合Python风格的实现思路，比你现在的写法要利落多啦！

先明确下咱们的核心需求：把原字典里「键指向一组值」的关系反转，变成「每个值元素作为新键，指向所有曾经引用过它的原键」，还要把原字典里没当过键但出现在值里的元素（比如例子里的4）也纳入新字典；原字典的键如果没被任何值引用，在新字典里就对应空集合。

方法一：逻辑清晰的高效实现（首推） #

这个写法兼顾可读性和执行效率，完全覆盖你的需求：

data = {"1": {"2", "3"}, "2": {"3", "4"}, "3": {"2", "4"}}

# 第一步：收集所有需要的键——原字典的键 + 所有值里的元素
all_keys = set(data.keys()).union(*data.values())
# 第二步：初始化新字典，每个键对应空集合
references = {key: set() for key in all_keys}
# 第三步：遍历原字典，把原键添加到对应值元素的集合里
for orig_key, orig_values in data.items():
    for val in orig_values:
        references[val].add(orig_key)

print(references)
# 输出：{'1': set(), '2': {'1', '3'}, '3': {'1', '2'}, '4': {'2', '3'}}

为啥说这个方法好用？

• 先一次性捞齐所有要用到的键，绝不会遗漏像4这种只在值里出现的元素
• 用字典推导式初始化，直接得到普通字典（不是defaultdict），完全匹配你要的输出格式
• 循环逻辑简单直接，没有多余的分支判断——如果原字典某个键的值是空集合，内层循环不会执行，新字典里对应的集合自然是空，完美贴合需求
• 时间复杂度是O(N+M)（N是原字典键数，M是所有值的元素总数），效率拉满

方法二：极致简洁的推导式写法（适合小数据量） #

如果你的数据量不大，追求代码极致简洁，可以用字典推导式+集合推导式的组合：

data = {"1": {"2", "3"}, "2": {"3", "4"}, "3": {"2", "4"}}

all_keys = set(data.keys()).union(*data.values())
references = {
    key: {orig_key for orig_key, orig_vals in data.items() if key in orig_vals}
    for key in all_keys
}

这个写法拆成多行是为了可读性，实际可以压缩成一行，非常Pythonic，但要注意：它的时间复杂度是O(K*N)（K是总键数），数据量大的时候效率不如第一种方法，适合小数据集用。

对比你当前的写法 #

你原来的代码里有个else分支，逻辑是「如果原键的值为空，就把这个键加到所有references的集合里」，这其实和你的需求不符哦——看你的示例输出，原键1没有被任何值引用，新字典里对应的集合是空，而不是加到所有其他集合里。上面的两种方法都自动处理了这种情况，不需要额外的分支判断，逻辑更严谨。

备注：内容来源于stack exchange，提问作者khteh