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

Question

I have a DataFrame with shape of (418, 13) and I want to just copy the two columns into a new DataFrame for outputting to a csv file. (I am writing a prediction)

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

However when I look into the outputted csv file I see this:

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

instead of what I expected which is:

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

Does anyone know why my code doesn't work? Thanks in advance.

Answer 1

无需创建新的DF：

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