python - 如何反转字典项并列出按公共(public)值分组的键

Question

这个问题在这里已经有了答案:





Dictionary comprehension for swapping keys/values in a dict with multiple equal values

(3 个回答)


2年前关闭。




我有一本字典，我想按常见值进行分组:

init_dict = {'00001': 'string1', '00002': 'string2', '00003': 'string1', '00004': 'string3', '00005': 'string2'}

我想创建一个新字典，对值进行分组并列出如下键:

new_dict = {'string1': ['00001', '00003'], 'string2':['00002', '00004'], 'string3': ['00004']}

我尝试了很多东西，这是我能得到的最接近的。

lookup = 'string1'
all_keys = []
for k, v in init_dict.items():
  if v == lookup:
    all_keys.append(k)
print(all_keys)

这将产生第一个列表:['00001', '00003']所以我想我可以以某种方式遍历 lookup 的列表值，但不能，因为我正在使用字符串。有没有办法做到这一点，有没有一种相对有效的方法，因为我的初始字典中有 53,000 个项目。任何帮助将不胜感激，因为我已经尝试了几个小时不同的事情。

Answer 1

使用 defaultdict , 指定 list作为默认参数，并从字典中附加相应的值:

from collections import defaultdict

d =  defaultdict(list)
for k,v in init_dict.items():
    d[v].append(k)

print(d)

defaultdict(list,
            {'string1': ['00001', '00003'],
             'string2': ['00002', '00005'],
             'string3': ['00004']})