python - pandas, groupby.agg 以给定的列名返回

Question

目前，我是

df.groupby('A')['B'].agg({'total' : np.sum})

这样聚合列就有一个列名 total。但是最近我收到警告

FutureWarning: using a dict on a Series for aggregation is deprecated and will be removed in a future version

那么现在首选的方式是什么？

Answer 1

我认为您需要在reset_index 中rename 或定义name 参数:

同时检查 deprecate groupby agg with a dictionary when renaming .

1.

df = df.groupby('A')['B'].sum().reset_index(name='total')

2.

df = df.groupby('A', as_index=False)['B'].sum().rename(columns={'B':'total'})

3.

df = df.groupby('A').agg({'B' : 'sum'}).rename(columns={'B':'total'}).reset_index()