python - 如何在 Python 中基于 groupby 添加一列的值？

Question

假设我有以下数据框：

year count 2001 14 2004 16 2001 2 2005 21 2001 22 2004 14 2001 8

我想按列分组并为每个给定年份year添加列。count我希望我的结果是

year count 2001 46 2004 30 2005 21

我正在努力寻找一种方法来做到这一点，有人可以帮忙吗？

标签： pythondataframe

Answer 1

import pandas as pd

df = pd.read_csv("test.csv")
df['count'] = pd.to_numeric(df['count'])
#df['count'] = df.groupby(['year'])['count'].sum()

total = df.groupby(['year'])['count'].sum()

print(total)

产量：

year
2001    46
2004    30
2005    21