在groupby操作期间重命名pandas数据框中的列

Question

我试图重命名作为groupby＆count操作的结果而获得的列，如下所示：

dfa = df.groupby('Product_ID').Product_ID.count().rename(columns={0: "Product",1:"Sale_count"}).reset_index()
print(dfa[:1])

获得的输出是

 Product_ID     0
0  P00000142  1130

列名不是我指定的。所以我使用下面的命令再次更改了它

dfa.columns =['product','sales']
print(dfa[:1])

 product  sales
0  P00000142   1130

然后我得到了预期的列名。但是我相信它将在第一个方法dataframe.rename本身中获得。第一个代码段dfa = df.groupby('Product_ID').Product_ID.count().rename(columns={0: "Product",1:"Sale_count"}).reset_index()中有什么问题，我没有得到预期的输出。

Answer 1

如评论中所述，您需要：

df.groupby('Product_ID').size().reset_index(name='sales_count')