如何用仅含单层for循环的字典推导式实现反向一对多映射

结论 #

可以实现，借助itertools.groupby预分组后即可使用仅含单层for循环的字典推导式完成反转操作，输出结果和原有两种方案完全一致。

实现代码 #

依赖导入 #

from itertools import groupby

核心实现 #

lst = [{'272': '4', '273': '4', '274': '4', '275': '5'}]
# 字典推导式仅含外层1个for循环，内层为分组内元素处理的列表推导式
dct = {val: [kv[0] for kv in group] 
       for val, group in groupby(sorted(lst[0].items(), key=lambda x: x[1]), key=lambda x: x[1])}

输出验证 #

print(dct)
# 运行输出：{'4': ['272', '273', '274'], '5': ['275']}

补充说明 #

• 由于itertools.groupby仅能聚合连续的相同key元素，所以需要先对原字典的键值对按值排序，保证相同值的键值对连续排列
• 该方案的时间复杂度为O(n log n)（主要来自排序步骤），性能低于普通for循环的O(n)方案，更适合小数据量场景使用
• 若不允许导入任何外部模块，纯原生Python语法无法仅靠单层for循环的字典推导式实现该效果，因为原生字典推导式单次迭代只能生成一个键值对，无法直接完成相同值的多键聚合操作

内容的提问来源于stack exchange，提问作者Engr Tanveer sultan