python讀取、寫入csv資料:csv模組
阿新 • • 發佈:2018-12-31
通過csv模組可以輕鬆讀取格式為csv的檔案,而且csv模組是python內建的,不需要下載就可以直接用。
一、準備csv檔案
檔名是 e:\t.csv,檔案內容:
org_id,org_name,state,emp_id
1,銷售1,'1',123
2,銷售2,'0',321
3,銷售3,'1',231
1,,'1',1234
二、讀取csv資料
程式碼非常簡單:
# -*- coding:gbk -*- import csv,os ##資料檔案 E:\t.csv ##org_id,org_name,state,emp_id ##1,銷售1,'1',123 ##2,銷售2,'0',321 ##3,銷售3,'1',231 ##1,,'1',1234 #拼接檔案路徑,os.path.sep是路徑分隔符 sFilename = 'e:'+os.path.sep+'t.csv' #開啟檔案 eFile = open(sFilename) #讀取csv檔案 eReader=csv.reader(eFile) #遍歷csv物件獲取資料,每一條資料都是一個list,每一列是list中的一個元素 #line_num是行號,這裡只讀取前100行 for row in eReader: if eReader.line_num <= 100: print('行 '+str(eReader.line_num) + ': '+str(row)) #列印行號 else: break #關閉檔案 eFile.close()
輸出:
三、寫入csv資料
輸出:# -*- coding:gbk -*- import csv,os sFilename = os.path.join('E:','r.csv') eFile = open(sFilename,'w',newline='') eWriter = csv.writer(eFile,delimiter=',',lineterminator='\r\n') eWriter.writerow(['org_id','org_name','state','emp_id']) eWriter.writerow([123,'銷售10','1',1111]) eWriter.writerow([1234,'銷售16','0',11115]) eFile.close()