按多个键分组并汇总/平均字典列表的多个值

Question

我是 Python 新手，在使用以下代码时遇到了问题。

我正在寻找一种方法来按多个键进行分组，并在 Python 中汇总/平均字典列表的值。下面的代码（也来自上一个问题/回复位于此处：Group by multiple keys and summarize/average values of a list of dictionaries）让我走上了正确的轨道，但我遇到了在循环中添加更多字段聚合的问题。

假设我有一个字典列表，如下所示：

input = [
{'msn': '001', 'source': 'foo', 'status': '1', 'qty': 100, 'vol': 100},
{'msn': '001', 'source': 'bar', 'status': '2', 'qty': 200, 'vol': 200},
{'msn': '001', 'source': 'foo', 'status': '1', 'qty': 300, 'vol': 300},
{'msn': '002', 'source': 'baz', 'status': '2', 'qty': 400, 'vol': 100},
{'msn': '002', 'source': 'baz', 'status': '1', 'qty': 500, 'vol': 400},
{'msn': '002', 'source': 'qux', 'status': '1', 'qty': 600, 'vol': 100},
{'msn': '003', 'source': 'foo', 'status': '2', 'qty': 700, 'vol': 200}]

到目前为止我的代码：

for key, grp in groupby(sorted(dict_list, key = grouper), grouper):
    temp_dict = dict(zip(["msn", "source"], key))
    temp_dict["qty"] = sum(item["qty"] for item in grp)
    temp_dict["vol"] = sum(item["vol"] for item in grp)
    result.append(temp_dict)

预期结果是：

{'msn': '001', 'source': 'foo', 'qty': 400, 'vol': 400},
{'msn': '001', 'source': 'bar', 'qty': 200, 'vol': 200},
{'msn': '002', 'source': 'baz', 'qty': 200, 'vol': 500},
{'msn': '003', 'source': 'foo', 'qty': 900, 'vol': 200}]

在 for 循环中放置 temp_dict["vol"] = sum(item["vol"] for item in grp) 不会产生所需的结果，这最终是我的问题。

如何在将另一个字段及其计算值添加（附加）到列表中时保留键、在代码中看到的分组？

提前感谢您的帮助。

Answer 1

如果您想多次迭代，您需要“复制”grp，itertools.tee 可以为您完成此操作

for key, grp in groupby(sorted(dict_list, key = grouper), grouper):
    temp_dict = dict(zip(["msn", "source"], key))
    grp1, grp2 = tee(grp)
    temp_dict["qty"] = sum(item["qty"] for item in grp1)
    temp_dict["vol"] = sum(item["vol"] for item in grp2)
    result.append(temp_dict)