2017-03-08 56 views
0

我有一個形狀爲(418,13)的DataFrame,我想將兩列複製到一個新的DataFrame中輸出到一個csv文件。 (我寫的預測)從一個DataFrame複製兩列到熊貓的另一個

csv_pred = prediction[["PassengerId", "Survived"]].copy() 
csv_pred.to_csv('n.csv') 

然而,當我看着輸出的CSV文件,我看到這一點:

,PassengerId,Survived 
0,892,0 
1,893,1 
2,894,0 
. . . 
. . . 

,而不是這是我所期待的:

"PassengerId","Survived" 
892,0 
893,1 
894,0 

不任何人都知道我的代碼爲什麼不起作用?提前致謝。

+1

輸出CSV文件中的第一列是你的數據框的默認指標。如果你不想索引列,在df.to_csv()函數中設置'index = False',如果你也不希望列標籤設置'header = False' –

回答

4

有沒有必要建立一個新的DF:

prediction[["PassengerId", "Survived"]].to_csv('/path/to/file.csv', index=False) 
相關問題