我正在使用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标头名称添加为值,并从标头名称中删除。这背后的原因是什么,是否有修复?
答案 0 :(得分:5)
将as_index参数设置为False。
data.groupby(['Store Number'], as_index=False).mean()
答案 1 :(得分:0)
@cel在运行groupby()后添加data = data.reset_index()
的评论中解决后添加了标题