1. 程式人生 > >python讀取、寫入csv資料:csv模組

python讀取、寫入csv資料:csv模組

通過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()
輸出: