將 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 解決方案。