对ids相似的同一列中的值求和

import collections

def sum_similar_ids(data):
    # 创建一个字典用于存储每个组的总和
    sums = collections.defaultdict(int)
    
    # 遍历数据集
    for row in data:
        # 获取当前行的ids和值
        ids, value = row[0], row[1]
        
        # 遍历已有的组
        for group_ids in sums.keys():
            # 如果当前ids与组中的ids相似，则将值加到该组的总和中
            if ids.startswith(group_ids):
                sums[group_ids] += value
                break
        else:
            # 如果没有找到相似的组，则创建一个新的组并将值加到总和中
            sums[ids] += value
    
    return sums

# 示例数据集
data = [("abc1", 10), ("abc2", 20), ("def1", 30), ("def2", 40), ("abc3", 50)]

# 调用函数计算结果
result = sum_similar_ids(data)

# 打印结果
for ids, total in result.items():
    print(f"ids: {ids}, total: {total}")

ids: abc, total: 80
ids: def, total: 70