使用Python對Csv文件操作實例代碼

2019-11-25 16:11:03

字體：大中小

來源：轉載

供稿：網友

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類型。這種方法要事先知道列的序號，比如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)

得到：