十分鐘教會你用Python處理CSV文件

更新時間：2022年06月11日 11:36:40 作者：程序員倉庫

大家都知道使用csv文件可以較容易地存儲多行且列相同的數(shù)據(jù),便于數(shù)據(jù)的讀取與解析,也常用于自動化測試過程中的數(shù)據(jù)參數(shù)化,下面這篇文章主要給大家介紹了關于如何利用Python處理CSV文件的相關資料,需要的朋友可以參考下

前言

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

A：”誒，老王，你看Python開發(fā)這么方便，以后會不會用到嵌入式設備？“

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

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

今天，作者就來總結一下使用Python處理物聯(lián)網行業(yè)中一種常見的文件格式：csv文件。

Python庫：csv

Python中集成了專用于處理csv文件的庫，名為：csv。

csv 庫中有4個常用的對象：

csv.reader：以列表的形式返回讀取的數(shù)據(jù)。
csv.writer：以列表的形式寫入數(shù)據(jù)。
csv.DictReader：以字典的形式返回讀取的數(shù)據(jù)。
csv.DictWriter：以字典的形式寫入數(shù)據(jù)。

讀取csv文件

假設現(xiàn)在要讀取的csv文件內容如下：

可以看到，該文件的第一行表明數(shù)據(jù)類型，我們暫且稱之為header。從第2行開始，保存的是設備采集到的數(shù)據(jù)。

使用csv.reader讀取數(shù)據(jù)

# 導入 csv 庫
import csv

# 以讀方式打開文件
with open("data.csv", mode="r", encoding="utf-8-sig") as f:
    
    # 基于打開的文件，創(chuàng)建csv.reader實例
    reader = csv.reader(f)

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

    # 逐行獲取數(shù)據(jù)，并輸出
    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讀取數(shù)據(jù)

# 導入 csv 庫
import csv

# 打開文件
with open("data.csv", encoding="utf-8-sig", mode="r") as f:

    # 基于打開的文件，創(chuàng)建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文件

假設我們現(xiàn)在要創(chuàng)建一個csv文件，將數(shù)據(jù)保存為如下形式：

使用csv.writer寫入數(shù)據(jù)

# 導入 csv 庫
import csv

# 創(chuàng)建列表，保存header內容
header_list = ["設備編號", "溫度", "濕度", "轉速"]

# 創(chuàng)建列表，保存數(shù)據(jù)
data_list = [
    [0, 31, 20, 1000],
    [1, 30, 22, 998],
    [2, 32, 33, 1005]
]

# 以寫方式打開文件。注意添加 newline=""，否則會在兩行數(shù)據(jù)之間都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基于打開的文件，創(chuàng)建 csv.writer 實例
    writer = csv.writer(f)

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

    # 寫入數(shù)據(jù)。
    # writerows() 一次寫入多行。
    writer.writerows(data_list)

使用csv.DictWriter寫入數(shù)據(jù)

# 導入 csv 庫
import csv

# 創(chuàng)建 header 列表
header_list = ["設備編號", "溫度", "濕度", "轉速"]

# 創(chuàng)建數(shù)據(jù)列表，列表的每個元素都是字典
data_list = [
    {"設備編號": "0", "溫度": 31, "濕度": 20, "轉速": 1000},
    {"設備編號": "1", "溫度": 30, "濕度": 22, "轉速": 998},
    {"設備編號": "2", "溫度": 32, "濕度": 23, "轉速": 1005},
]

# 以寫方式打開文件。注意添加 newline=""，否則會在兩行數(shù)據(jù)之間都插入一行空白。
with open("new_data.csv", mode="w", encoding="utf-8-sig", newline="") as f:
    
    # 基于打開的文件，創(chuàng)建 csv.DictWriter 實例，將 header 列表作為參數(shù)傳入。
    writer = csv.DictWriter(f, header_list)

    # 寫入 header
    writer.writeheader()

    # 寫入數(shù)據(jù)
    writer.writerows(data_list)

關于寫入，需要注意：