为什么只有最后一个条目与 itemgetter 和 groupby 一起返回

Question

我想知道下面的代码返回的内容是否符合预期？

我在 Python 3.7 上运行它。

我期望“预期”输出，但只返回每个组的最后一个条目，如“实际”中所示。

import operator
import itertools

adict = {'a': 1, 'b':3, 'c': 1, 'd': 2, 'e': 3, 'f': 1}

dict((i, dict(v)) for i, v in itertools.groupby(adict.items(), operator.itemgetter(1))) 

Expected Output: {1: {'a': 1, 'c': 1, 'f':1}, 2: {'d': 2}, 3: {'b', 'e'} }

Actual : {1: {'f': 1}, 3: {'e': 3}, 2: {'d': 2}}

Answer 1

如果你打破dict((i, dict(v)) for i, v in itertools.groupby(adict.items(), operator.itemgetter(1)))，应该清楚为什么只有最后一个键值对在最终字典中：

for i, v in itertools.groupby(adict.items(), operator.itemgetter(1)):
    print(i, dict(v))

输出

1 {'a': 1}
3 {'b': 3}
1 {'c': 1}
2 {'d': 2}
3 {'e': 3}
1 {'f': 1}

在每一个上调用dict 将用相同的外键覆盖之前创建的dict。

可以通过使用defaultdict(dict) 来修复它，同时确保update 现有的内部字典具有新的键值对：

from collections import defaultdict

adict = {'a': 1, 'b': 3, 'c': 1, 'd': 2, 'e': 3, 'f': 1}

output = defaultdict(dict)
for i, v in adict.items():
    output[v][i] = v
print(output)

输出

defaultdict(<class 'dict'>, {1: {'a': 1, 'c': 1, 'f': 1}, 3: {'b': 3, 'e': 3}, 2: {'d': 2}})

您不必使用defaultdict，但它会保存存在检查：

adict = {'a': 1, 'b': 3, 'c': 1, 'd': 2, 'e': 3, 'f': 1}

output = {}
for i, v in adict.items():
    if v in output:
        output[v][i] = v
    else:
        output[v] = {i: v}
print(output)

输出

{1: {'a': 1, 'c': 1, 'f': 1}, 3: {'b': 3, 'e': 3}, 2: {'d': 2}}

正如 Jon 在 cmets 中提到的，这也可以通过首先按相同键 (operator.itemgetter(1)) 对 adict.values 进行排序来解决：

dict((i, dict(v)) for i, v in itertools.groupby(sorted(adict.items(), key=itemgetter(1)), operator.itemgetter(1)))

itertools.groupbydocumentation中实际上提到了这个警告：

... 一般情况下，iterable 需要已经在同一个 key 函数上排序。