仅将两列从一个 DataFrame 复制到 Pandas 中的另一列

Question

我有一个形状为 (418, 13) 的 DataFrame，我只想将两列复制到一个新的 DataFrame 中以输出到 csv 文件。（我在写预测）

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

有谁知道为什么我的代码不起作用？提前致谢。

Answer 1

无需创建新的 DF：

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