Python 读取CSV文件

csv文件如下所示：

 

按行读取csv文件

（1）reader函数

import csv
with open('test.csv','rt') as csvfile: #此csv文件是一个文本文件，并非二进制文件，如是二进制文件，'rt'改为'rb'
  reader = csv.reader(csvfile)
  rows = [row for row in reader]
print (rows)

   结果：

  

   如果想读取特定的行： 

import csv
with open('test.csv','rt') as csvfile:
	reader = csv.reader(csvfile)
	for i,rows in enumerate(reader):
		if i == 2:
			row = rows
print (row)

   结果：

   ['3', '4', '5', '3', '9']

（2）DictReader

csv文件：

 

 和reader函数类似，接收一个可迭代的对象，能返回一个生成器，但是返回的每一个单元格都放在一个字典的值内，而这个字典的键则是这个单元格的标题（即列头）。用下面的代码可以看到DictReader的结构：

import csv
with open('test.csv','rt') as csvfile:
	reader = csv.DictReader(csvfile)
	rows = [row for row in reader]
print (rows)

结果：

[OrderedDict([('实验', '8'), ('观测', '2'), ('模拟', '7'), ('误差', '6'), ('总计', '8')]),

OrderedDict([('实验', '3'), ('观测', '4'), ('模拟', '5'), ('误差', '3'), ('总计', '9')]),

OrderedDict([('实验', '0'), ('观测', '0'), ('模拟', '0'), ('误差', '0'), ('总计', '0')]),

OrderedDict([('实验', '0'), ('观测', '7'), ('模拟', '8'), ('误差', '9'), ('总计', '9')]),

OrderedDict([('实验', '0'), ('观测', '0'), ('模拟', '0'), ('误差', '0'), ('总计', '0')]),

OrderedDict([('实验', '0'), ('观测', '0'), ('模拟', '0'), ('误差', '0'), ('总计', '0')]),

OrderedDict([('实验', '0'), ('观测', '0'), ('模拟', '0'), ('误差', '0'), ('总计', '0')])]

如果我们想用DictReader读取csv的某一列，就可以用列的标题查询：

import csv
with open('test.csv','rt') as csvfile:
	reader = csv.DictReader(csvfile)
	for row in reader:
		if row['误差']=='9':
			print(row)

结果：

OrderedDict([('实验', '0'), ('观测', '7'), ('模拟', '8'), ('误差', '9'), ('总计', '9')])

可见，DictReader很适合读取csv的的行（记录）。