CSV 本质上是文本文件,能提供与 Excel 的表格类似的二维数据结构。因为是文本文件,读写都比 Excel 方便容易的多,经常拿来作为数据的中转格式。本文讨论如何快速有效的按列读取 CSV 文件。
假设“A.csv“文件内容是:
No.,Name,Age,Score
1,Apple,12,98
2,Ben,13,97
3,Celia,14,96
4,Dave,15,95
用 Excel 打开这个 csv, 看起来是这样的:
利用Python自带的 csv 模块 ,有两种方法可以提取其中的一列:
方法一 reader函数
第一种方法使用reader函数,接收一个可迭代的对象(比如csv文件),能返回一个生成器,就可以从其中解析出csv的内容:比如下面的代码可以读取csv的全部内容,以行为单位:
import csv
with open('A.csv','rb') as csvfile:
reader = csv.reader(csvfile)
rows= [row for row in reader]
print rows
得到:
[['No.', 'Name', 'Age', 'Score'],
<