如何对多个字段分组的数量求和? [英] How to sum the quantity grouped by more than one field?

问题描述

如何对按 [pid][sid][eid][dbid] 字段分组的 'qty' 字段求和:

How do i sum the 'qty' field grouped by [pid][sid][eid][dbid] fields:

d = [
   {'pid': 146, 'sid': 6, 'eid': 6212, 'dbid': 1, 'qty': 10},
   {'pid': 146, 'sid': 6, 'eid': 6212, 'dbid': 1, 'qty': 20},
   {'pid': 232, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 1},
   {'pid': 232, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 1},
   {'pid': 146, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 100},
   {'pid': 146, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 100},
   ]

# non-working code
sum_of_qty = []
for row in d:
    sum_of_qty[[row['pid']][row['sid']][row['eid']][row['dbid']]] += row['qty'] 

这是我想要的最终结果:

This is the final result i want:

sum_of_qty = [
    {'pid': 146, 'sid': 6, 'eid': 6121, 'dbid': 1, 'qty': 30},
    {'pid': 232, 'sid': 56, 'eid': 6121, 'dbid': 1, 'qty': 2},
    {'pid': 146, 'sid': 56, 'eid': 6121, 'dbid': 1, 'qty': 200}
    ]

推荐答案

你可以使用itertools.groupby:

import itertools
d = [
  {'pid': 146, 'sid': 6, 'eid': 6212, 'dbid': 1, 'qty': 10},
  {'pid': 146, 'sid': 6, 'eid': 6212, 'dbid': 1, 'qty': 20},
  {'pid': 232, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 1},
  {'pid': 232, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 1},
  {'pid': 146, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 100},
  {'pid': 146, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 100},
]
new_d = [list(b) for _, b in itertools.groupby(d, key=lambda x:[x['pid'], x['sid']])]
final_result = [{**i[0], **{'qty':sum(b['qty'] for b in i)}} for i in new_d]

输出:

[{'pid': 146, 'sid': 6, 'eid': 6212, 'dbid': 1, 'qty': 30}, {'pid': 232, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 2}, {'pid': 146, 'sid': 56, 'eid': 6212, 'dbid': 1, 'qty': 200}]

这篇关于如何对多个字段分组的数量求和?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！