当前位置: 首页 > 面试题库 >

将Python字典键分组为一个列表,并使用该列表作为值创建一个新字典

姜景辉
2023-03-14
问题内容

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


问题答案:

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}}

(对设置值的输出进行排序是一个巧合,是如何实现整数的哈希值的副作用;集合是无序结构)。



 类似资料: