Home / Questions / 136787
Accepted
Yoel Macia Delgado
Asked: 2020-02-08 02:58:38 +0800 CST 2020-02-08 02:58:38 +0800 CST

对具有相同键的字典列表的值求和

  • 772

我有一个字典列表

lst = [ {'x': 1, 'y': 2}, {'x': 1, 'y': 3 }, {'x': 2, 'y': 2 } ]

字典始终具有相同的键，并且值始终在变化。

所以我想在键“x”具有相同值时对键“y”的值求和。

输出应给出：lst # [{'x': 1, 'y': 5 }, {'x': 2, 'y': 2 }].

python

3 Answers

  1. Best Answer

    一种选择是itertools.groupby根据 key 的值对字典进行分组'x'。一旦完成，使用列表压缩来获得最终列表很简单，添加键的值'y'（以及其他键，如果它们存在）：

    import itertools
from operator import itemgetter



lst = [{'x': 1, 'y': 2}, {'x': 1, 'y': 3 }, {'x': 2, 'y': 2 }]

grupos = itertools.groupby(sorted(lst, key=itemgetter('x')), key=itemgetter('x'))

res = [{'x': v, 'y': sum(dicc['y'] for dicc in diccs)} for v, diccs in grupos]

print(res)

    输出是[{'x': 1, 'y': 5}, {'x': 2, 'y': 2}]。

    • 1

  2. 另一种解决方法：

    from collections import defaultdict

lst = [{'x': 1, 'y': 2}, {'x': 1, 'y': 3 }, {'x': 2, 'y': 2 }]

aux = defaultdict(int)
for d in lst:
    aux[d['x']] += d['y']

res = [{'x':k, 'y':v} for (k,v) in res.items()]
    • 1

  3. 我已经解决了这个问题，就像这样。

    lst = [ {'x': 1, 'y': 2}, {'x': 1, 'y': 3 }, {'x': 2, 'y': 2 } ]

result_list = []
name_dict = {}
for dict_item in lst:
    x = dict_item['x']
    y = dict_item['y']
    if x in name_dict:
        pos = name_dict[x]
        result_list[pos] = {'x':x, 'y': (result_list[pos]['y']+y)}
    else:
        result_list.append(dict_item)
        name_dict[x] = len(result_list) - 1

print result_list
    • 0