字典列表的聚合

Question

我正在尝试对我在我正在处理的 django 应用程序中找到的这些数据进行排序和聚合。问题是我迷失了遍历列表和存储数据的最佳方式。

这是我所拥有的一个例子：

from score.models import LocData

q = [

        {'ref': '002', 'loc': 'seattle', 'total': '200'}, 

        {'ref': '002', 'loc': 'seattle', 'total': '100'}, 

        {'ref': '123', 'loc': 'dallas', 'total': '100'}, 

        {'ref': '452', 'loc': 'cleveland', 'total': '600'}, 

        {'ref': '123', 'loc': 'dallas', 'total': '200'}, 

        {'ref': '002', 'loc': 'seattle', 'total': '300'}

        ]

我想最终得到的是下面的列表，它由 ref 字段聚合并使用 loc 维护此字段，但添加了 total 字段。这是所需的输出。

q = [

        {'ref': '002', 'loc': 'seattle', 'total': '600'}, 

        {'ref': '123', 'loc': 'dallas', 'total': '300'}, 

        {'ref': '452', 'loc': 'cleveland', 'total': '600'}, 

        ]

有人能告诉我我有哪些工具可以做到这一点吗？提前致谢！

Answer 1

可以先构造一个中间字典，然后根据所需的输出对其进行转换：from collections import defaultdictq = [        {'ref': '002', 'loc': 'seattle', 'total': '200'},         {'ref': '002', 'loc': 'seattle', 'total': '100'},         {'ref': '123', 'loc': 'dallas', 'total': '100'},         {'ref': '452', 'loc': 'cleveland', 'total': '600'},         {'ref': '123', 'loc': 'dallas', 'total': '200'},         {'ref': '002', 'loc': 'seattle', 'total': '300'}    ]temp_dict = defaultdict(int)for entry in q:    temp_dict[(entry['ref'], entry['loc'])] += int(entry['total'])result = [{'ref': k[0], 'loc': k[1], 'total': str(v)} for k, v in temp_dict.items()]print(result)# [{'ref': '002', 'loc': 'seattle', 'total': '600'},#  {'ref': '123', 'loc': 'dallas', 'total': '300'},#  {'ref': '452', 'loc': 'cleveland', 'total': '600'}]

Answer 2

您可以使用一个聚合collections.Counter：from collections import Counterfrom pprint import pprintq = [    {"ref": "002", "loc": "seattle", "total": "200"},    {"ref": "002", "loc": "seattle", "total": "100"},    {"ref": "123", "loc": "dallas", "total": "100"},    {"ref": "452", "loc": "cleveland", "total": "600"},    {"ref": "123", "loc": "dallas", "total": "200"},    {"ref": "002", "loc": "seattle", "total": "300"},]counts = Counter()for x in q:    ref, loc, total = x["ref"], x["loc"], x["total"]    counts[ref, loc] += int(total)pprint(    [        {"ref": ref, "loc": loc, "total": str(total)}        for (ref, loc), total in counts.items()    ])#[{'loc': 'seattle', 'ref': '002', 'total': '600'},# {'loc': 'dallas', 'ref': '123', 'total': '300'},# {'loc': 'cleveland', 'ref': '452', 'total': '600'}]