以下是一个示例代码,演示了如何按组汇总值并保留原始数据:
import pandas as pd
# 创建示例数据
data = {
'Group': ['A', 'A', 'B', 'B', 'B'],
'Value': [1, 2, 3, 4, 5]
}
df = pd.DataFrame(data)
# 按组计算求和
df['Sum'] = df.groupby('Group')['Value'].transform('sum')
print(df)
输出结果:
Group Value Sum
0 A 1 3
1 A 2 3
2 B 3 12
3 B 4 12
4 B 5 12
在这个示例中,我们首先创建了一个包含'Group'和'Value'列的DataFrame。然后,我们使用groupby
方法按'Group'列进行分组,并使用transform
方法计算每个组的总和。最后,我们将计算得到的总和添加到原始DataFrame中的新列'Sum'中。这样,我们就得到了按组汇总值并保留原始数据的结果。