python二维数组分组合计 itertools groupby
提供简单思路
from itertools import groupby
a = [['a', 'b', 1, 2, 3],
['a', 'c', 1, 2, 3],
['a', 'd', 1, 2, 3],
['b', 'b', 1, 2, 3],
['b', 'c', 1, 2, 3],
['a', 'e', 1, 2, 3],
]
a.sort(key=lambda x: x[0])
lstg = groupby(a, lambda x: x[0])
result = []
for key, group in lstg:
temp_list = list(group)
temp = list(zip(*temp_list))
sum_line = list(map(sum, temp[2:]))
sum_line.insert(0, '')
sum_line.insert(0, temp[0][0])
print(sum_line)
result.extend(temp_list)
result.extend([sum_line])
total = list(map(sum, list(zip(*a))[2:]))
total.insert(0, '')
total.insert(0, '合计')
result.append(total)
print(result)
result = [['a', 'b', 1, 2, 3],
['a', 'c', 1, 2, 3],
['a', 'd', 1, 2, 3],
['a', 'e', 1, 2, 3],
['a', '', 4, 8, 12],
['b', 'b', 1, 2, 3],
['b', 'c', 1, 2, 3],
['b', '', 2, 4, 6],
['合计', '', 6, 12, 18]]
