PANDAS GroupBy删除标题

Question

我正在使用PANDAS groupBy并注意到它正在删除我正在运行它的值的标题名称.

data = pd.read_csv("<CSV FILE NAME>", low_memory=False)
print data.head()
print data.columns

给我以下输出:

        Store ID        Daily Sales
0       4444444         436
1       4555555          406
2       6435353         487
3       3421456          637
4       1111111         516
Index([u'Store ID', u' Daily Sales'], dtype='object')

我跑的时候

data = data.groupby(['Store Number']).mean()
print data.head()
print data.columns

输出更改为

                  Daily Sales
Store ID             
4166646        236.280394
4166663        152.061884
4166664        131.163746
4166665        144.920044
4166666        225.075027
Index([u'Daily Sales'], dtype='object')

Store ID标头名称将作为值添加,并从标头名称中删除.这背后的原因是什么,是否有修复？

Answer 1

将as_index参数设置为False.

data.groupby(['Store Number'], as_index=False).mean()