python二维数组分组合计 itertools groupby

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]]