python - 将 Python 字典键分组为列表，并以此列表为值创建一个新字典

Question

我有一本 python 字典

d = {1: 6, 2: 1, 3: 1, 4: 9, 5: 9, 6: 1}

由于上述字典中的值不是唯一的，我想将唯一值的所有键分组为一个列表并创建一个新字典，如下所示:

v = {6:[1], 1:[2, 3, 6], 9: [4, 5]}

注意新字典v的键应该排序。我发现很难想象和实现这个字典创建。

Answer 1

使用 collections.defaultdict为了方便:

from collections import defaultdict

v = defaultdict(list)

for key, value in sorted(d.items()):
    v[value].append(key)

但你也可以使用标准的 dict 来做到这一点，使用 dict.setdefault() :

v = {}

for key, value in sorted(d.items()):
    v.setdefault(value, []).append(key)

以上排序键first;稍后对输出字典的值进行排序更加麻烦和低效。

如果有人不需要需要对输出进行排序，您可以放弃 sorted() 调用，并使用 sets(键在输入字典中保证是唯一的，所以不会丢失任何信息):

v = {}

for key, value in d.items():
    v.setdefault(value, set()).add(key)

生产:

{6: {1}, 1: {2, 3, 6}, 9: {4, 5}}

(集合值的输出排序是巧合，是整数哈希值实现方式的副作用；集合是无序结构)。