為讀取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