以下是一个示例代码,演示如何按列进行累计总和分组:
import pandas as pd
# 创建示例数据
data = {
'组': ['A', 'A', 'B', 'B', 'A', 'B'],
'数值': [1, 2, 3, 4, 5, 6]
}
df = pd.DataFrame(data)
# 按组进行累计总和分组
df['累计总和'] = df.groupby('组')['数值'].cumsum()
print(df)
输出结果如下:
组 数值 累计总和
0 A 1 1
1 A 2 3
2 B 3 3
3 B 4 7
4 A 5 8
5 B 6 13
在这个示例中,我们使用了groupby
函数来按组进行分组,并使用cumsum
函数计算累积总和。最后,我们将累计总和存储在新的一列中。