python - 如何在Python中计算多列的累计总和

Question

我有一个数据集，例如，

data=pd.DataFrame({'id':pd.Series([1,1,1,2,2,3,3,3]),'var1':pd.Series([1,2,3,4,5,6,7,8]),'var2':pd.Series([11,12,13,14,15,16,17,18]),
'var3':pd.Series([21,22,23,24,25,26,27,28])})

这里我需要根据id计算所有列(var1，var2，var3)的分组累积和。如何编写 python 代码来根据我的要求生成输出？

提前致谢。

Answer 1

如果我理解正确，您可以使用 DataFrame.groupby 来计算按 'id' 列分组的各列的累积总和。像这样的东西:

import pandas as pd
data=pd.DataFrame({'id':[1,1,1,2,2,3,3,3],'var1':[1,2,3,4,5,6,7,8],'var2':[11,12,13,14,15,16,17,18], 'var3':[21,22,23,24,25,26,27,28]})
data.groupby('id').apply(lambda x: x.drop('id', axis=1).cumsum(axis=1).sum())