使用Python對Csv文件操作

csv是Comma-Separated Values的縮寫，是用文本文件形式儲存的表格數據，比如如下的表格：

就可以存儲為csv文件，文件內容是：

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

假設上述csv文件保存為"test.csv"

1.讀文件

如何用Python像操作Excel一樣提取其中的一列，即一個字段，利用Python自帶的csv模塊，有兩種方法可以實現：

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

#!/usr/bin/python3
# -*- coding: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
# -*- coding: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的結構：

# -*- coding: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
# -*- coding: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
# -*- coding: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)

得到：