时间戳

首页 > 解决方案 > 基于ID的累积SUM

python - 基于ID的累积SUM

问题描述

问题：

我有一个数据框，我从带有熊猫的“csv”文件导入，看起来像这样（简化示例）：

id  amount 
1   50     
1   10     
1    5      
2   10     
2   15     
2   25     
2   40

所以想法是将金额相加并将结果添加到具有相同 id 的所有记录中，如下所示：

 id  amount total
1   50     65 
1   10     65 
1   5      65
2   10     90
2   15     90
2   25     90
2   40     90

因此，生成的数据框应该包含一个额外的列，总计对应于每个 id 的数量，但保留原始数据框，为了使其正常工作，每个 Id 不可能有一条记录与数量。

我正在使用熊猫并操作 csv 文件。

谢谢！

标签： pythonpandas

解决方案

利用pandas.join

df.join(df.groupby('id').amount.sum(), rsuffix='_', on = 'id')

输出：

   id  amount  amount_
0   1      50       65
1   1      10       65
2   1       5       65
3   2      10       90
4   2      15       90
5   2      25       90
6   2      40       90

甚至更好（感谢WB）：

df.assign(total = df.groupby('id').amount.transform('sum'))

输出

   id  amount  total
0   1      50     65
1   1      10     65
2   1       5     65
3   2      10     90
4   2      15     90
5   2      25     90
6   2      40     90

推荐阅读