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)

参考:https://blog.csdn.net/bxprog/article/details/47806409

猜你喜欢

转载自www.cnblogs.com/aaronhoo/p/9232979.html