WeHelp
Python 如何處理 CSV 格式檔案
2022-08-26 02:48:26
CSV (Comma-Separated Values) 格式的文字檔案,一直是用來表達表格、二維資料的好方法。由於格式本身很單純,沒有任何額外的附加功能,所以相容性也非常高,可以被幾乎任何試算表工具 ( Excel, Google Sheets, ...) 開啟,並做後續的資料操作,因此 CSV 是一個相當好的資料交換格式。 以下簡單的介紹我們如何利用 Python 程式,將資料以 CSV 的格式儲存到檔案中,或者是從 CSV 格式的檔案中取出資料。 --- ### 什麼是 CSV 格式 首先,我們先了解什麼是 CSV 格式,所謂 CSV 格式檔案,顧名思義,**就是每行字都使用逗號隔開的表格式資料**: ``` 資料,資料,資料,... 資料,資料,資料,... ``` 以下為 3x3 的資料範例,表達總共有三列,每列有三個欄位的表格式資料。 ``` 1,2,3 4,5,6 7,8,9 ``` --- ### 儲存 CSV 格式的檔案 那麼,我們如何利用 Python 程式,建立一個以 CSV 格式儲存資料的檔案呢?請參考以下步驟: 1. 載入 Python 內建的 csv 模組 2. 開啟想要儲存資料的檔案 3. 建立 CSV Writer 物件 4. 使用 CSV Writer 物件的 writerow 方法,將每行的資料以 List 的型態寫入檔案中 以下程式會將上述 3x3 的資料範例,寫入 data.csv 的檔案: ``` python import csv with open("data.csv", "w", newline="") as csvfile: writer = csv.writer(csvfile) writer.writerow([1, 2, 3]) writer.writerow([4, 5, 6]) writer.writerow([7, 8, 9]) ``` 順利的話,用文字編輯器打開 data.csv 檔案,就會順利看到儲存在檔案中的 CSV 格式資料嘍! --- ### 讀取 CSV 格式的檔案 反過來,如果我們要讀取 CSV 格式的檔案,要如何進行呢?請參考以下步驟: 1. 載入 Python 內建的 csv 模組 2. 開啟想要讀取資料的檔案 3. 建立 CSV Reader 物件 4. 使用 CSV Reader 物件搭配 for 迴圈,將每行的資料以 List 的型態取出 以下程式會將我們剛建立好的 data.csv 檔案資料,讀取出來後印在終端機中: ``` python import csv with open("data.csv", "r", newline="") as csvfile: reader = csv.reader(csvfile) for row in reader: print(row) # row 是 List 的型態,可以用 print(row[0], row[1], row[2]) 分別取得印出 ``` --- 是不是很簡單呢?