将 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)

但是你也可以使用 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}}

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

Answer 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)))

当然，如果你真的想，你可以用它来构造一个 dict：

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

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