Python：使用list / dict理解來模仿groupby的效果

Question

我需要轉換字典列表：

original = [
    {'type': 'a', 'length': 34, 'width': 74},
    {'type': 'a', 'length': 15, 'width': 22},
    {'type': 'b', 'length': 53, 'width': 54},
    {'type': 'b', 'length': 11, 'width': 45},
] 

轉換為以key type為鍵的值的字典：

expected = {
    'a': [
        {'type': 'a', 'length': 34, 'width': 74},
        {'type': 'a', 'length': 15, 'width': 22},
    ],
    'b': [
        {'type': 'b', 'length': 53, 'width': 54},
        {'type': 'b', 'length': 11, 'width': 45},
    ],
}  

這可以通過itertools.groupby或通過手動遍歷列表來實現，但是itertools.groupby列表/字典理解有什么方法嗎？

Answer 1

您可以執行以下操作：

{t: [i for i in original if i['type'] == t] for t in {i['type'] for i in original}}

但它既難以閱讀，又具有最壞的運行時復雜度O（n²） ，其中n是列表中的項目數。 在排序列表上使用itertools.groupby既快捷又容易閱讀。