使用 Python 處理 CSV 檔案

nisan發表於2024-07-27

使用 Python 處理 CSV 檔案

簡介

這份文件將講解如何使用 Python 程式語言來處理 CSV 檔案。CSV 檔案是一種常用的資料儲存格式,它將資料以逗號分隔的形式儲存在文字檔案中。Python 擁有強大的庫可以輕鬆地讀取、寫入和操作 CSV 檔案。

依賴庫

為了處理 CSV 檔案,你需要使用 Python 的 csv 庫。該庫提供了一系列函式,可以幫助你讀取、寫入和操作 CSV 檔案。你可以透過以下程式碼匯入 csv 庫:

import csv

讀取 CSV 檔案

使用 csv 庫的 reader 物件可以讀取 CSV 檔案的內容。reader 物件會將檔案內容解析成一個二維列表,其中每一行表示 CSV 檔案中的一個資料行,每一列表示資料行中的一個值。

with open('data.csv', 'r') as csvfile:
    reader = csv.reader(csvfile)
    for row in reader:
        print(row)

這段程式碼將讀取 data.csv 檔案的內容,並列印每一行資料。

寫入 CSV 檔案

可以使用 csv 庫的 writer 物件將資料寫入 CSV 檔案。writer 物件可以將二維列表或其他可迭代物件寫入 CSV 檔案。

with open('output.csv', 'w', newline='') as csvfile:
    writer = csv.writer(csvfile)
    writer.writerow(['姓名', '年齡', '城市'])
    writer.writerows([['張三', 25, '北京'], ['李四', 30, '上海']])

這段程式碼將建立一個名為 output.csv 的 CSV 檔案,並在檔案中寫入三行資料。

操作 CSV 檔案

除了讀取和寫入之外,還可以使用 csv 庫對 CSV 檔案進行其他操作,例如:

  • 跳過頭部行: 使用 next(reader) 方法可以跳過 CSV 檔案中的第一行,通常這行是標題行。
  • 指定分隔符: 使用 delimiter 引數可以指定 CSV 檔案中的分隔符。例如,csv.reader(csvfile, delimiter=';') 會將分號作為分隔符。
  • 指定引號字元: 使用 quotechar 引數可以指定 CSV 檔案中的引號字元。例如,csv.writer(csvfile, quotechar='|') 會使用豎線作為引號字元。

總結

Python 的 csv 庫提供了豐富的功能,可以幫助你輕鬆地處理 CSV 檔案。透過使用 readerwriter 物件,你可以讀取、寫入和操作 CSV 檔案中的資料。希望這份文件能幫助你更好地理解和使用 Python 處理 CSV 檔案。

相關文章