52

我有一本 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的键应该被排序。我发现很难想象和实现这个字典的创建。请建议我一种简单有效的方法。

4

2 回答 2

80

使用方便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)

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

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

v = {}

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

生产：

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

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

于 2013-04-01T21:12:53.713 回答
25

dict如果您在一天结束时实际上不需要 a ，您可以使用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 解决方案。

于 2013-04-01T21:24:09.040 回答