Python 讀取與寫入 CSV 檔案教學與範例

這裡介紹如何在 Python 中使用 csv 模組,讀取與寫入逗點分隔檔。

逗點分隔(Comma-Separated Values,簡稱 csv)是一種簡單的文字檔格式,以逗號分隔不同欄位的資料,很多軟體在儲存與交換表格資料時都支援這樣的格式。

在 Python 中若要讀取或產生 csv 的檔案,可以使用內建的 csv 模組,以下是使用方式以及範例程式碼。

讀取 CSV 檔案

假設我們有一個 csv 檔案 iris.csv,其內容如下:

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 檔的內容讀取出來:

import csv

# 開啟 CSV 檔案
with open('iris.csv', newline='') as csvfile:

  # 讀取 CSV 檔案內容
  rows = csv.reader(csvfile)

  # 以迴圈輸出每一列
  for row in rows:
    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 這個檔案,資料如下:

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
[略]

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

import csv
with open('/etc/passwd', newline='') as csvfile:

  # 以冒號分隔欄位,讀取檔案內容
  rows = csv.reader(csvfile, delimiter=':')

  for row in rows:
    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 格式,這樣在存取各定資料時會方便一些:

import csv
with open('iris.csv', newline='') as csvfile:

  # 讀取 CSV 檔內容,將每一列轉成一個 dictionary
  rows = csv.DictReader(csvfile)

  # 以迴圈輸出指定欄位
  for row in rows:
    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 檔案,可以使用以下這段程式碼:

import csv

# 開啟輸出的 CSV 檔案
with open('output.csv', 'w', newline='') as csvfile:
  # 建立 CSV 檔寫入器
  writer = csv.writer(csvfile)

  # 寫入一列資料
  writer.writerow(['姓名', '身高', '體重'])

  # 寫入另外幾列資料
  writer.writerow(['令狐沖', 175, 60])
  writer.writerow(['岳靈珊', 165, 57])

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

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

指定分隔字元

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

import csv
with open('output.csv', 'w', newline='') as csvfile:

  # 以空白分隔欄位,建立 CSV 檔寫入器
  writer = csv.writer(csvfile, delimiter=' ')

  writer.writerow(['姓名', '身高', '體重'])
  writer.writerow(['令狐沖', 175, 60])
  writer.writerow(['岳靈珊', 165, 57])

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

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

一次寫入二維表格

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

import csv

# 二維表格
table = [
    ['姓名', '身高', '體重'],
    ['令狐沖', 175, 60],
    ['岳靈珊', 165, 57]
]

with open('output.csv', 'w', newline='') as csvfile:
  writer = csv.writer(csvfile)

  # 寫入二維表格
  writer.writerows(table)

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

寫入 Dictionary

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

import csv
with open('output.csv', 'w', newline='') as csvfile:
  # 定義欄位
  fieldnames = ['姓名', '身高', '體重']

  # 將 dictionary 寫入 CSV 檔
  writer = csv.DictWriter(csvfile, fieldnames=fieldnames)

  # 寫入第一列的欄位名稱
  writer.writeheader()

  # 寫入資料
  writer.writerow({'姓名': '令狐沖', '身高': 175, '體重': 60})
  writer.writerow({'姓名': '岳靈珊', '身高': 165, '體重': 57})

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

參考資料:Python 2 DocumentPython 3 Document

程式設計

3 Comments

  1. DT

    請問一下,我將要載入的csv檔放置在跟python檔同一個資料夾下,但是卻出現FileNotFoundError的情況,要如何解決此問題呢?

    File “C:\Users\user\Desktop\reyo python\csv_reader.py”, line 3, in
    with open(‘result.csv’, newline=”) as csvfile:
    FileNotFoundError: [Errno 2] No such file or directory: ‘result.csv’

  2. Passby

    試試看”\result.csv”
    或是放絕對路徑
    “C:\Users\user\Desktop\reyo python\result.csv”

  3. Raines

    嗨,我使用下面的code
    不過一直出現
    TypeError: ‘_csv.reader’ object is not callable
    請問一下是甚麼問題

    import csv

    # 開啟 CSV 檔案
    with open(‘d:/Users/ChenR/Desktop/HHP/1810C35.dat’, newline=”) as csvfile:

    # 讀取 CSV 檔案內容
    rows = csv.reader(csvfile)

    # 以迴圈輸出每一列
    for row in rows:
    print(row)

Leave a Reply