Python如何對Csv檔案操作的執行個體代碼分享

這篇文章主要介紹了使用Python對Csv檔案操作執行個體代碼，非常具有實用價值，需要的朋友可以參考下

csv是Comma-Separated Values的縮寫，是用文字檔形式儲存的表格式資料，比如如下的表格：

就可以儲存為csv檔案，檔案內容是：

No.,Name,Age,Score1,mayi,18,992,jack,21,893,tom,25,954,rain,19,80

假設上述csv檔案儲存為"test.csv"

1.讀檔案

如何用Python像操作Excel一樣提取其中的一列，即一個欄位，利用Python內建的csv模組，有兩種方法可以實現：

第一種方法使用reader函數，接收一個可迭代的對象（比如csv檔案），能返回一個產生器，就可以從其中解析出csv的內容：比如下面的代碼可以讀取csv的全部內容，以行為單位：

#!/usr/bin/python3# -*- conding:utf-8 -*-author = 'mayi'import csv#讀with open("test.csv", "r", encoding = "utf-8") as f:  reader = csv.reader(f)  rows = [row for row in reader]print(rows)

得到：

[['No.', 'Name', 'Age', 'Score'], ['1', 'mayi', '18', '99'], ['2', 'jack', '21', '89'], ['3', 'tom', '25', '95'], ['4', 'rain', '19', '80']]

要提取其中某一列，可以用下面的代碼：

#!/usr/bin/python3# -*- conding:utf-8 -*-author = 'mayi'import csv#讀取第二列的內容with open("test.csv", "r", encoding = "utf-8") as f:  reader = csv.reader(f)  column = [row[1] for row in reader]print(column)

得到：

['Name', 'mayi', 'jack', 'tom', 'rain']

注意從csv讀出的都是str類型。這種方法要事Crowdsourced Security Testing道列的序號，比如Name在第2列，而不能根據'Name'這個標題查詢。這時可以採用第二種方法：

第二種方法是使用DictReader，和reader函數類似，接收一個可迭代的對象，能返回一個產生器，但是返回的每一個儲存格都放在一個字典的值內，而這個字典的鍵則是這個儲存格的標題（即列頭）。用下面的代碼可以看到DictReader的結構：

# -*- conding:utf-8 -*-author = 'mayi'import csv#讀with open("test.csv", "r", encoding = "utf-8") as f:  reader = csv.DictReader(f)  column = [row for row in reader]print(column)

得到：

[{'No.': '1', 'Age': '18', 'Score': '99', 'Name': 'mayi'}, {'No.': '2', 'Age': '21', 'Score': '89', 'Name': 'jack'}, {'No.': '3', 'Age': '25', 'Score': '95', 'Name': 'tom'}, {'No.': '4', 'Age': '19', 'Score': '80', 'Name': 'rain'}]

如果我們想用DictReader讀取csv的某一列，就可以用列的標題查詢：

#!/usr/bin/python3# -*- conding:utf-8 -*-author = 'mayi'import csv#讀取Name列的內容with open("test.csv", "r", encoding = "utf-8") as f:  reader = csv.DictReader(f)  column = [row['Name'] for row in reader]print(column)

得到：

['mayi', 'jack', 'tom', 'rain']

2.寫檔案

讀檔案時，我們把csv檔案讀入列表中，寫檔案時會把列表中的元素寫入到csv檔案中。

#!/usr/bin/python3# -*- conding:utf-8 -*-author = 'mayi'import csv#寫：追加row = ['5', 'hanmeimei', '23', '81']out = open("test.csv", "a", newline = "")csv_writer = csv.writer(out, dialect = "excel")csv_writer.writerow(row)

得到：