使用 Python Pandas 写入 to_csv：选择要插入新数据的列索引

Question

我的程序中有一组数据输出，我想写入 .csv 文件。我可以使用旧输入数据创建一个新文件，然后是右侧最后一列中的新数据。如何操作我的输出数据到哪一列？另外，如何选择不在新文件中包含旧输入数据？我是熊猫新手。

谢谢！

Answer 1

从文件加载：

import pandas as pd
df = pd.read_csv('D:\\Apps\\Coursera\\Kaggle-Titanic\\Data\\train.csv', header = 0)

一些操作：

df['Gender'] = df.Sex.map(lambda x: 0 if x=='female' else 1)
df['FamilySize'] = df.SibSp + df.Parch

复制一些字段到新的：

result = df[['Sex', 'Survived', 'Age']]

删除不需要的字段：

del result['Sex']

保存到文件：

result.to_csv('D:\\Apps\\Coursera\\Kaggle-Titanic\\Swm\\result.csv', index=False)

或者，如果您只想保存某些字段或按特定顺序：

df[['Sex', 'Survived', 'Age']].to_csv('D:\\Apps\\Coursera\\Kaggle-Titanic\\Swm\\result.csv', index=False)