Python DictReader讀寫csv檔案
阿新 • • 發佈:2019-02-15
準備
此文件的示例程式碼是基於python3.5寫的。
使用csv庫前,先匯入csv庫:
import csv
讀取內容
假設csv檔案的內容如下圖所示,DictReader會將第一行的內容(類標題)作為key值,第二行開始才是資料內容。即圖中的csv檔案有2列7行資料,第一列的key值為id,第二列的key值為class:
with open(‘name.csv’) as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
#迴圈列印資料的id和class值,此迴圈執行7次
print(row[‘id’],row[‘class’])
列印結果為:
0 5
1 4
2 13
…
6 15
寫入內容
現假設我們要建立一個格式如上圖所示的csv檔案,則實現程式碼應該如下:
with open(‘name.csv’,’w’) as csvfile:
writer = csv.DictWriter(csvfile,fieldnames=[‘id’,’class’])
#寫入列標題,即DictWriter構造方法的fieldnames引數
writer.writeheader()
for data in datas:
writer.writerow({‘id’:data[0],’class’:data[1]})