十分鐘教會你用Python處理CSV檔案

2022-06-11 14:00:15

前言

在前幾年，如果你和嵌入式開發人員推薦Python，大概會是這樣一種場景：

A：”誒，老王，你看Python開發這麼方便，以後會不會用到嵌入式裝置？“

B：“別做夢了，那玩意兒速度賊慢，肯定滿足不了效能要求…”

但近幾年，隨著半導體行業的迅猛發展，嵌入式處理器的效能有了很大幅度的提高。與此同時，Python語言本身也在不斷優化。因此，在嵌入式領域引入Python已經成為了必然趨勢。

今天，作者就來總結一下使用Python處理物聯網行業中一種常見的檔案格式：csv檔案。

Python庫：csv

Python中整合了專用於處理csv檔案的庫，名為：csv。

csv 庫中有4個常用的物件：

csv.reader：以列表的形式返回讀取的資料。
csv.writer：以列表的形式寫入資料。
csv.DictReader：以字典的形式返回讀取的資料。
csv.DictWriter：以字典的形式寫入資料。

讀取csv檔案

假設現在要讀取的csv檔案內容如下：

可以看到，該檔案的第一行表明資料型別，我們暫且稱之為header。從第2行開始，儲存的是裝置採集到的資料。

使用csv.reader讀取資料

# 匯入 csv 庫
import csv

# 以讀方式開啟檔案
with open("data.csv", mode="r", encoding="utf-8-sig") as f:
    
    # 基於開啟的檔案，建立csv.reader範例
    reader = csv.reader(f)

    # 獲取第一行的header
    # header[0] = "裝置編號"
    # header[1] = "溫度"
    # header[2] = "溼度"
    # header[3] = "轉速"
    header = next(reader)

    # 逐行獲取資料，並輸出
    for row in reader:
        print("{}{}: {}={}, {}={}, {}={}".format(header[0], row[0],
                                                 header[1], row[1],
                                                 header[2], row[2],
                                                 header[3], row[3]))

程式執行結果如下：

裝置編號0: 溫度=31, 溼度=20, 轉速=1000
裝置編號1: 溫度=30, 溼度=22, 轉速=998
裝置編號2: 溫度=32, 溼度=23, 轉速=1005

使用csv.DictReader讀取資料

# 匯入 csv 庫
import csv

# 開啟檔案
with open("data.csv", encoding="utf-8-sig", mode="r") as f:

    # 基於開啟的檔案，建立csv.DictReader範例
    reader = csv.DictReader(f)

    # 輸出資訊
    for row in reader:
        print("裝置編號{}: 溫度={}, 溼度={}, 轉速={}".format(row["裝置編號"],
                                                   row["溫度"],
                                                   row["溼度"],
                                                   row["轉速"]))

程式執行結果如下：

裝置編號0: 溫度=31, 溼度=20, 轉速=1000
裝置編號1: 溫度=30, 溼度=22, 轉速=998
裝置編號2: 溫度=32, 溼度=23, 轉速=1005

寫入csv檔案

假設我們現在要建立一個csv檔案，將資料儲存為如下形式：

使用csv.writer寫入資料

# 匯入 csv 庫
import csv

# 建立列表，儲存header內容
header_list = ["裝置編號", "溫度", "溼度", "轉速"]

# 建立列表，儲存資料
data_list = [
    [0, 31, 20, 1000],
    [1, 30, 22, 998],
    [2, 32, 33, 1005]
]

# 以寫方式開啟檔案。注意新增 newline=""，否則會在兩行資料之間都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基於開啟的檔案，建立 csv.writer 範例
    writer = csv.writer(f)

    # 寫入 header。
    # writerow() 一次只能寫入一行。
    writer.writerow(header_list)

    # 寫入資料。
    # writerows() 一次寫入多行。
    writer.writerows(data_list)

使用csv.DictWriter寫入資料

# 匯入 csv 庫
import csv

# 建立 header 列表
header_list = ["裝置編號", "溫度", "溼度", "轉速"]

# 建立資料列表，列表的每個元素都是字典
data_list = [
    {"裝置編號": "0", "溫度": 31, "溼度": 20, "轉速": 1000},
    {"裝置編號": "1", "溫度": 30, "溼度": 22, "轉速": 998},
    {"裝置編號": "2", "溫度": 32, "溼度": 23, "轉速": 1005},
]

# 以寫方式開啟檔案。注意新增 newline=""，否則會在兩行資料之間都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基於開啟的檔案，建立 csv.DictWriter 範例，將 header 列表作為引數傳入。
    writer = csv.DictWriter(f, header_list)

    # 寫入 header
    writer.writeheader()

    # 寫入資料
    writer.writerows(data_list)

關於寫入，需要注意：