快捷導航

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

更新時間：2017年05月12日 10:28:33 作者：騎著螞蟻流浪

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

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

就可以存儲為csv文件，文件內(nèi)容是：

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

假設(shè)上述csv文件保存為"test.csv"

1.讀文件

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

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

#!/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

#讀取第二列的內(nèi)容
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列，而不能根據(jù)'Name'這個標題查詢。這時可以采用第二種方法：

第二種方法是使用DictReader，和reader函數(shù)類似，接收一個可迭代的對象，能返回一個生成器，但是返回的每一個單元格都放在一個字典的值內(nèi)，而這個字典的鍵則是這個單元格的標題（即列頭）。用下面的代碼可以看到DictReader的結(jié)構(gòu)：

# -*- 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列的內(nèi)容
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)

得到：