pandas - df.columns 未列出的所有列名

Question

我想在我的数据帧上执行 groupby 和 agg 功能，所以我执行了以下代码

basic_df = df.groupby(['S2PName','S2PName-Category'], sort=False)['S2PGTotal'].agg([('totSale','sum'), ('count','size')])

basic_df.head(2)

我的运单：

                              totSale               count
S2PName     S2PName-Category        
IDLY         Food                 598771.47             19749
DOSA          Food                567431.03             14611

现在我尝试使用打印列basic_df.columns

我的运单：

Index(['totSale', 'count'], dtype='object')

为什么没有显示其他两列“S2pname 和 S2PName-category”。我还需要做什么才能显示它们？

谢谢 ！

标签： pandas

Answer 1

在末尾添加as_index=False, 或reset_index()

basic_df = df.groupby(['S2PName','S2PName-Category'], sort=False,as_index=False)['S2PGTotal'].agg([('totSale','sum'), ('count','size')])

#basic_df = df.groupby(['S2PName','S2PName-Category'], sort=False)['S2PGTotal'].agg([('totSale','sum'), ('count','size')]).reset_index()