1. 程式人生 > >Python DictReader讀寫csv檔案

Python DictReader讀寫csv檔案

準備

此文件的示例程式碼是基於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]})