使用 Python 將 CSV 資料載入到列表和字典中


簡介

CSV 檔案格式是計算機之間交換資料的常用格式。功能強大且適應性強的程式語言 Python 內建了許多庫,可以讀取和寫入 CSV 檔案。列表和字典是兩種最流行的 Python 資料結構,它們非常適合儲存 CSV 資料以進行進一步處理和分析。

本文將逐步引導您瞭解如何使用 Python 將 CSV 資料載入到列表和字典中,並使用實際示例使過程更容易理解。

將 CSV 資料載入到 Python 列表中

Python 附帶的 csv 模組允許使用者讀取和寫入 CSV 檔案。我們將使用一個名為 data.csv 的虛構 CSV 檔案來演示如何將 CSV 資料載入到列表中。它包含以下資訊

Name, Age, Country
Alice, 25, USA
Bob, 30, UK
Charlie, 35, Canada

讓我們看看如何將此 CSV 資料新增到列表中

import csv

with open('data.csv', 'r') as file:
   reader = csv.reader(file)
   data_list = list(reader)
    
print(data_list)

此指令碼讀取 CSV 資料並將其轉換為列表的列表,其中每個子列表對應於 CSV 檔案中的一行

[['Name', 'Age', 'Country'], ['Alice', '25', 'USA'], ['Bob', '30', 'UK'], ['Charlie', '35', 'Canada']]

將 CSV 資料載入到 Python 字典中

Python 字典可以提供一種更具組織性和適應性強的儲存 CSV 資料的方法。csv.DictReader 物件用於將 CSV 資料匯入字典,它將每一行視為具有標題的排序字典對映。

我們應該將我們的 data.csv 檔案載入到字典中

import csv

with open('data.csv', 'r') as file:
   reader = csv.DictReader(file)
   data_dict = [row for row in reader]

print(data_dict)

執行此指令碼時,會生成一個字典列表,每個字典對應於 CSV 檔案中的一行。

[{'Name': 'Alice', 'Age': '25', 'Country': 'USA'}, {'Name': 'Bob', 'Age': '30', 'Country': 'UK'}, {'Name': 'Charlie', 'Age': '35', 'Country': 'Canada'}]

這裡,CSV 標題(姓名、年齡和國家/地區)充當字典的鍵,而字典中的其餘條目充當匹配的值。

使用列表和字典儲存 CSV 資料的優勢

將 CSV 資料載入到列表或字典中是儲存和管理資料的一種簡單方法。例如,您可以對每個資料條目應用函式,或者迴圈遍歷列表或字典以檢索特定資料條目。

此外,使用字典可以提供其他好處。對於某些使用場景,鍵值對可能是訪問資料的更邏輯方法。例如,您可以立即從 data_dict 中檢索第二人的“國家/地區”,如下所示

print(data_dict[1]['Country'])  # Prints 'UK'

結論

Python 中內建的 csv 模組使處理 CSV 檔案變得簡單。您是選擇將 CSV 資料儲存在列表中還是字典中,主要取決於您擁有的資料型別以及您打算如何處理和使用它。與字典相比,列表很簡單,並提供有序的專案集合,而字典則提供結構化且靈活的方法,併為輕鬆訪問資料而將鍵對映到值。

如果您想使用 Python 分析和處理資料,那麼瞭解這些 CSV 資料處理技術至關重要。透過實際實現和分析,您可以成功地為您的特定用例選擇最佳方法,從而建立易讀且高效的程式碼。無論您是在進行資料分析、機器學習還是網路抓取,這些技能都將非常有用。

更新於:2023-07-18

4K+ 瀏覽量

開啟您的 職業生涯

透過完成課程獲得認證

開始學習
廣告

© . All rights reserved.