[ Python 文章收集 ] Python 讀取與寫入 CSV 檔案教學與範例

Source From Here 
Preface 
這裡介紹如何在 Python 中使用 csv 模組，讀取與寫入逗點分隔檔。 逗點分隔（Comma-Separated Values，簡稱 csv）是一種簡單的文字檔格式，以逗號分隔不同欄位的資料，很多軟體在儲存與交換表格資料時都支援這樣的格式。在 Python 中若要讀取或產生 csv 的檔案，可以使用內建的 csv 模組，以下是使用方式以及範例程式碼。 

讀取 CSV 檔案 
假設我們有一個 csv 檔案 iris.csv，其內容如下： 

view plaincopy to clipboardprint?

1. sepal_length,sepal_width,petal_length,petal_width,species  
2. 5.1,3.5,1.4,0.2,setosa  
3. 4.9,3,1.4,0.2,setosa  
4. 4.7,3.2,1.3,0.2,setosa  
5. 4.6,3.1,1.5,0.2,setosa  

我們可以使用下面這段程式碼將這個 csv 檔的內容讀取出來： 

view plaincopy to clipboardprint?

1. import csv  
2.   
3. # 開啟 CSV 檔案  
4. with open('iris.csv', newline='') as csvfile:  
5.   
6.   # 讀取 CSV 檔案內容  
7.   rows = csv.reader(csvfile)  
8.   
9.   # 以迴圈輸出每一列  
10.   for row in rows:  
11.     print(row)  

我們使用 csv.reader 讀取出來的 rows 會是一個二維的 list，裡面就是整張表格的資料，這裡我們把每一列的 list 直接輸出，執行後會像這樣： 

['sepal_length', 'sepal_width', 'petal_length', 'petal_width', 'species']
['5.1', '3.5', '1.4', '0.2', 'setosa']
['4.9', '3', '1.4', '0.2', 'setosa']
['4.7', '3.2', '1.3', '0.2', 'setosa']
['4.6', '3.1', '1.5', '0.2', 'setosa']

這裡在開啟 csv 檔案時加上了 newline='' 參數，這是為了讓資料中包含的換行字元可以正確被解析，所以建議在讀取 csv 檔案時都固定加入這個參數。 

指定分隔字元 
如果資料欄位之間的分隔字元不是使用預設的逗號，而是其他字元的話，在讀取時就要自行指定欄位的分隔字元。假設我們要讀取 Linux 的 /etc/passwd 這個檔案，資料如下： 

view plaincopy to clipboardprint?

1. root:x:0:0:root:/root:/bin/bash  
2. daemon:x:1:1:daemon:/usr/sbin:/usr/sbin/nologin  
3. bin:x:2:2:bin:/bin:/usr/sbin/nologin  
4. sys:x:3:3:sys:/dev:/usr/sbin/nologin  
5. sync:x:4:65534:sync:/bin:/bin/sync  

這個檔案內容的各個欄位是以冒號（:）來分隔的，我們可以使用以下這段程式碼來讀取： 

view plaincopy to clipboardprint?

1. import csv  
2. with open('/etc/passwd', newline='') as csvfile:  
3.   
4.   # 以冒號分隔欄位，讀取檔案內容  
5.   rows = csv.reader(csvfile, delimiter=':')  
6.   
7.   for row in rows:  
8.     print(row)  

輸出如下： 

['root', 'x', '0', '0', 'root', '/root', '/bin/bash']
['daemon', 'x', '1', '1', 'daemon', '/usr/sbin', '/usr/sbin/nologin']
['bin', 'x', '2', '2', 'bin', '/bin', '/usr/sbin/nologin']
['sys', 'x', '3', '3', 'sys', '/dev', '/usr/sbin/nologin']
['sync', 'x', '4', '65534', 'sync', '/bin', '/bin/sync']

讀取成 Dictionary 
我們也可以將 csv 檔案的內容讀取進來之後，轉為 Python 的 dictionary 格式，這樣在存取各定資料時會方便一些： 

view plaincopy to clipboardprint?

1. import csv  
2. with open('iris.csv', newline='') as csvfile:  
3.   
4.   # 讀取 CSV 檔內容，將每一列轉成一個 dictionary  
5.   rows = csv.DictReader(csvfile)  
6.   
7.   # 以迴圈輸出指定欄位  
8.   for row in rows:  
9.     print(row['sepal_length'], row['species'])  

這個例子中我們使用以 csv.DictReader 來讀取 CSV 檔案的內容，它會自動把第一列（row）當作欄位的名稱，將第二列以後的每一列轉為 dictionary，這樣我們就可以使用欄位的名稱來存取資料。執行後的輸出會像這樣： 

5.1 setosa
4.9 setosa
4.7 setosa
4.6 setosa
5 setosa

寫入 CSV 檔案 
如果我們在程式中產生了表格的資料，想要儲存為 csv 檔案，可以使用以下這段程式碼： 

view plaincopy to clipboardprint?

1. import csv  
2.   
3. # 開啟輸出的 CSV 檔案  
4. with open('output.csv', 'w', newline='') as csvfile:  
5.   # 建立 CSV 檔寫入器  
6.   writer = csv.writer(csvfile)  
7.   
8.   # 寫入一列資料  
9.   writer.writerow(['姓名', '身高', '體重'])  
10.   
11.   # 寫入另外幾列資料  
12.   writer.writerow(['令狐沖', 175, 60])  
13.   writer.writerow(['岳靈珊', 165, 57])  

產生的 output.csv 檔案內容會像這樣： 

姓名,身高,體重
令狐沖,175,60
岳靈珊,165,57

指定分隔字元 
輸出 csv 檔案時也可以自行指定欄位的分隔字元，例如： 

view plaincopy to clipboardprint?

1. import csv  
2. with open('output.csv', 'w', newline='') as csvfile:  
3.   
4.   # 以空白分隔欄位，建立 CSV 檔寫入器  
5.   writer = csv.writer(csvfile, delimiter=' ')  
6.   
7.   writer.writerow(['姓名', '身高', '體重'])  
8.   writer.writerow(['令狐沖', 175, 60])  
9.   writer.writerow(['岳靈珊', 165, 57])  

輸出的 output.csv 檔案內容會變成這樣： 

姓名 身高 體重
令狐沖 175 60
岳靈珊 165 57

一次寫入二維表格 
如果我們的資料是已經整理好的二維表格，也可以一次就把整張表格寫進 csv 檔案中： 

view plaincopy to clipboardprint?

1. import csv  
2.   
3. # 二維表格  
4. table = [  
5.     ['姓名', '身高', '體重'],  
6.     ['令狐沖', 175, 60],  
7.     ['岳靈珊', 165, 57]  
8. ]  
9.   
10. with open('output.csv', 'w', newline='') as csvfile:  
11.   writer = csv.writer(csvfile)  
12.   
13.   # 寫入二維表格  
14.   writer.writerows(table)  

這樣產生的 csv 檔案也會跟上面的例子相同。 

寫入 Dictionary 
如果我們在 Python 中的資料格式是 dictionary，也可以使用 csv.DictWriter 直接將 dictionary 寫入 csv 檔案中： 

view plaincopy to clipboardprint?

1. import csv  
2. with open('output.csv', 'w', newline='') as csvfile:  
3.   # 定義欄位  
4.   fieldnames = ['姓名', '身高', '體重']  
5.   
6.   # 將 dictionary 寫入 CSV 檔  
7.   writer = csv.DictWriter(csvfile, fieldnames=fieldnames)  
8.   
9.   # 寫入第一列的欄位名稱  
10.   writer.writeheader()  
11.   
12.   # 寫入資料  
13.   writer.writerow({'姓名': '令狐沖', '身高': 175, '體重': 60})  
14.   writer.writerow({'姓名': '岳靈珊', '身高': 165, '體重': 57})  

此範例輸出的檔案內容也跟上面的例子相同。