2017-03-08 59 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) 
相关问题