python3 数据处理 将相同列的另一列数据相加

Question

有如下一个从oralce数据库中获取到的数据信息
[("张三"，"10","AAA"，"CCC"),("张三","20","BBB","CCC"),...]

如果各个元组中第一个数据是相同的话，将其第二列相加，合并成一列数据，其他列数据如果是相同的话就保留下来，如果是不相同的话就将该字段的数据舍弃，比如上面数据最终希望的结果是
[("张三"，"30","CCC"),..] (第三列数据一个是AAA一个是BBB不相同，就将其舍去不要)

请问下各位大佬有什么方法能进行这样的处理？ 我试过采用有序字典，但不需要相加的列就不好处理，就想请教下有没有其他合适的防范，谢谢

Answer 1

from functools import reduce

def merge(s):

    def _merge(a, b):
        i = a[-1]
        if i[0] == b[0]:
            a[-1] = (
                i[0],
                str(int(i[1])+int(b[1])),
                *[j[0] if j[0] == j[1] else None for j in zip(i[2:], b[2:])]
            )
        else:
            a.append(b)
        return a

    s[0] = [s[0]]
    return [tuple(j for j in i if j) for i in reduce(_merge, s)]

print(merge(s))