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

问题描述：

我有一本 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的键应该排序。我发现很难想象和实现这个字典的创建。

解决方案 1：

为方便使用collections.defaultdict：

from collections import defaultdict

v = defaultdict(list)

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

但您也可以使用 bog-standard 来实现dict，方法是dict.setdefault()：

v = {}

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

上述方法首先对键进行排序；稍后对输出字典的值进行排序会更加麻烦和低效。

如果有人不需要对输出进行排序，则可以放弃sorted()呼叫，并使用集合（输入字典中的键保证是唯一的，因此不会丢失任何信息）：

v = {}

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

生产：

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

（集合值的输出是排序的，这是一个巧合，是整数哈希值实现方式的副作用；集合是无序结构）。

解决方案 2：

如果你实际上并不需要dict在一天结束时，你可以使用itertools.groupby：

from itertools import groupby
from operator import itemgetter

for k, v in groupby(sorted(d.items(), key=itemgetter(1)), itemgetter(1)):
    print(k, list(map(itemgetter(0), v)))

当然，如果你真的愿意的话，你可以用它来构造一个字典：

{
    k: list(map(itemgetter(0), v))
    for k, v in groupby(sorted(d.items(), key=itemgetter(1)), itemgetter(1))
}

但此时，您最好使用 Martijn 的 defaultdict 解决方案。