2021-04-13

Pandas读写csv，并新增一列

为读取csv，并DataFrame增加一列，再自由组合列并保存到csv文件：

import pandas as pd

sourceFile='d:\person.csv' #person.csv包括id,name,age三个列
df=pd.read_csv(sourceFile,low_memory=False) #读取csv,设置low_memory=False防止内存不够时报警告
df['company']='google' #增加新的列company
#以下保存指定的列到新的csv文件，index=0表示不为每一行自动编号，header=1表示行首有字段名称
df.to_csv('d:\new_person.csv',columns=['id','name','company'],index=0,header=1)