矩阵中根据条件求和？

Question

矩阵如下：

1,2017,2,51,2018,1,51,2018,2,42,2017,1,51,2018,1,2

根据前三列分出三种，产生新的矩阵：

1,2017,2,51,2018,1,71,2018,2,42,2017,1,5

少了一行，是因为只要前三列相同，那第四列求和，2+5=7

如何计算？

Answer 1

就是分组求和的思想。m = [[1,2017,2,5],[1,2018,1,5],[1,2018,2,4],[2,2017,1,5],[1,2018,1,2]]# 先建立个dict,# 前三项作为key, 第四项作为valued = dict()for i in m:     d_key = tuple(i[:-1])     d_value = i[-1]    if d_key not in d:         d.update({d_key:d_value})    else:         d[d_key] += d_value          # 再把dict转为数组result = list()for d_key,d_value in d.items():     tmp = list(d_key)     tmp.append(d_value)     result.append(tmp) print(result)result就是：[[1, 2018, 2, 4], [1, 2018, 1, 7], [2, 2017, 1, 5], [1, 2017, 2, 5]]