用同一模式合并多个字典

问题描述 投票:0回答:2

假设我有一个长度未知、模式未知--但一致--的字典列表,如下所示。请注意,这些值总是相同长度的列表。

res = [{'x': [2, 1], 'v': [49280, 536704]},
 {'x': [1, 4], 'v': [12336, 340000]},
 {'x': [5, 6], 'v': [524360, 151624]},
 {'x': [7, 1], 'v': [94280, 81968]},
 {'x': [1, 1], 'v': [241856, 335904]},
 {'x': [7, 7], 'v': [463016, 598040]},
 {'x': [2, 9], 'v': [606256, 422016]},
 {'x': [1, 1], 'v': [151680, 1237120]}]

有没有一种有效的方法可以将这些值合并到一个字典中?这个例子的最终结果将是。

 {'x': [2, 1, 1, 4, 5, 6, 7, 1, 1, 1, 7, 7, 2, 9, 1, 1], 
  'v': [49280, 536704, 12336, 340000, 524360, 151624, 94280, 81968, 241856, 335904, 463016, 598040, 606256, 422016, 151680, 1237120]}

工作方案

这样做是可行的,但我想知道这种操作是否有单行本?

output_dct = {}
for dct in res:
    for k, v in dct.items():
        output_dct[k] = v if k not in output_dct.keys() else output_dct[k] + v
python dictionary
2个回答
1
投票

一个稍好的方法是使用 dict.setdefault

前。

output_dct = {}
for dct in res:
    for k, v in dct.items():
        output_dct.setdefault(k, []).extend(v)

或使用 collections.defaultdict

from collections import defaultdict
output_dct = defaultdict(list)
for dct in res:
    for k, v in dct.items():
        output_dct[k].extend(v)

print(output_dct)

输出。

{'x': [2, 1, 1, 4, 5, 6, 7, 1, 1, 1, 7, 7, 2, 9, 1, 1], 
'v': [49280, 536704, 12336, 340000, 524360, 151624, 94280, 81968, 241856, 335904, 463016, 598040, 606256, 422016, 151680, 1237120]}

1
投票

一句话,你说呢? 给理解力强的手机!

>>> {k: sum((d[k] for d in res), []) for d in res for k in d}
{'x': [2, 1, 1, 4, 5, 6, 7, 1, 1, 1, 7, 7, 2, 9, 1, 1], 'v': [49280, 536704, 12336, 340000, 524360, 151624, 94280, 81968, 241856, 335904, 463016, 598040, 606256, 422016, 151680, 1237120]}

(编辑:实际上,我可能会用一个 defaultdict 就像Rakesh所做的那样,因为嵌套理解在页面上很紧凑,但对其他代码读者来说不一定很明显)。)

© www.soinside.com 2019 - 2024. All rights reserved.