將標題添加到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