将标题添加到Panda的处理数据框中

Question

Here是我的数据集。

我正在使用.groupby和.size方法从以前的数据框（数据）创建一个新的熊猫数据框（ptocol），如下所示。这表现得如预期的那样，但结果是没有列标题的数据框。

我试了很长时间并且检查了讨论的解决方案here。但它不适合我。以下是我的代码。

import pandas as pd 
import numpy 

data = pd.read_csv('first.csv') 
ptocol = data.groupby(["Protocol"], as_index=False).size().rename(columns={0:'NumOfPackets'}) # dosn't work 
#ptocol = data.groupby(["Protocol"], as_index=False).count() #doesn't work 
print ptocol 
ptocol.to_csv('protocol.csv') 

实际结果（protocol.csv）：

0x200e,26 
ARP,100746 
ATMTCP,48 
BOOTP,123 
BZR,4 
... 
... 

预期的结果（protocol.csv）：

Protocol,NumOfPackets 
0x200e,26 
ARP,100746 
ATMTCP,48 
BOOTP,123 
BZR,4 
... 
... 

任何意见/建议，欢迎

Answer 1

.size()返回一系列对象，你可以使用reset_index()把它转换成一个数据帧，试试这个来代替：

ptocol = data.groupby("Protocol").size().rename('NumOfPackets').reset_index() 
ptocol.to_csv('protocol.cv', index = False) 

这给了这样的事情，不一样的数据，你的，但格式是你在找什么：

Symbol,NUM 
A,5 
AA,5 
AAAP,5