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)