數據編碼與處理-6.1 讀寫CSV檔案

Photo by Mika Baumeister on Unsplash

Q: 讀寫一個CSV文件。

Answer

對於大多數的 CSV 格式的數據讀寫問題，都可以使用 csv 這個套件，有人會說使用Pandas會更棒，沒錯但本刊以討論原生套件原主。假設目前有一個csv檔案(stocks.csv)有以下股票數據：

CSV範例數據

使用csv套件逐行讀取數據

在上面的程式碼中，row 會是一List。因此，爲了讀取某個欄位，你需要使用下標， 如 row[0] 讀取Symbol，row[4] 讀取Change。但是這種使用索引讀取特定數據的方式，很容易搞混或記錯，欄位一多你就會發瘋，所以可以使用 namedtuple 命名元組。舉例:

這樣的寫法允許你使用column名稱如 row.Symbol 和 row.Change 代替索引訪問。另一種做法是將資料以字典的方法讀入也看，舉例:

這樣可以也使用column名去讀取每一行的數據了，例如，row[‘Symbol’] 或 者 row[‘Change’]。

寫入CSV

如果你的資料是字典的話，請這樣做