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

寫入 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 檔案也會跟上面的例子相同。

from : https://blog.gtwang.org/programming/python-csv-file-reading-and-writing-tutorial/