Python中键的字典理解：[1,2,3] [重复]

Question

虽然我一直在提高我的 Python 技能，但我有一个问题。 我的代码如下：

# def invertDictionary(dict):
#     new_dict = {} 
#     for key, value in dict.items():
#                 if value in new_dict:
#                         new_dict[value].append(key) 
#                 else: 
#                     new_dict[value]=[key]
#     return new_dict

def invertDictionary(dict):
    new_dict = {value:([key] if value else [key]) for key, value in dict.items()}
    return new_dict;

invertDictionary({'a':3, 'b':3, 'c':3})

我正在尝试获得像 {3:['a','b','c']} 这样的输出。我已经使用普通的for循环实现了这一点；我只想知道如何使用 Dictionary Comprehension 获得这些结果。我试过但附加它得到一个错误。请让我知道如何实现这一目标。 提前致谢！

Answer 1

您错过了也需要列表理解来构建列表。
遍历 dict 中的值，并为每个值构建所需的键列表。

请注意，这是一个二次过程，而规范（且更易读）for 循环是线性的。

d = {'a':3, 'b':3, 'c':3, 'e':4, 'f':4, 'g':0}
inv_dict = {v: [key for key, val in d.items() if val == v]
               for v in set(d.values())}

结果：

{0: ['g'],
 3: ['a', 'b', 'c'],
 4: ['e', 'f']
}

这样可以吗？

Answer 2

虽然带有常规 for 循环的原始版本是解决此问题的最佳解决方案，但这里是 @Prune 答案的变体，不会多次遍历字典

>>> import itertools
>>> d = {'a':3, 'b':3, 'c':3, 'e':4, 'f':4, 'g':0}
>>> {group_key:[k for k,_ in dict_items] 
     for group_key,dict_items in itertools.groupby(
                                             sorted(d.items(),key=lambda x:x[-1]),
                                             key=lambda x:x[-1]
                                             )
      }
{0: ['g'], 3: ['a', 'b', 'c'], 4: ['e', 'f']}
>>> 

首先我们使用lambda function 使用lambda function 的键函数对dict 的项目按值排序，以提取项目元组的值部分，然后我们使用groupby 将具有相同值的那些分组加上相同的键功能，最后用列表理解提取键

--

正如 Kelly 所说，我们可以使用 dict 中的 get 方法来获取值以使其更短，并利用对 dict 的迭代为您提供其键的事实

>>> {k: list(g) for k, g in itertools.groupby(sorted(d, key=d.get), d.get)}
{0: ['g'], 3: ['a', 'b', 'c'], 4: ['e', 'f']}
>>> 

Answer 3

您可以使用defalutdict 和append 方法。

from collections import defaultdict

dict1 = {'a': 3, 'b': 3, 'c': 3}
dict2 = defaultdict(list)
{dict2[v].append(k) for k, v in dict1.items()}

dict2
>>> defaultdict(list, {3: ['a', 'b', 'c']})