python - PANDAS Group通过删除标题

Question

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

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 header 名称作为值添加并从 header 名称中删除。这背后的原因是什么？是否有解决办法？

Answer 1

将 as_index 参数设置为 False。

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