python csv文件头_我与python3擦肩而过（二）—— csv文件头哪去啦？

在看Python Data Visualization Cookbook 这本书(基于python2)，开始时读取csv文件头的时候出现问题。查了资料，又是python3的问题，从这个链接找到答案。

书中源码是这样的：

importcsv

filename= 'ch02-data.csv'data=[]try:

with open(filename) as f:

reader=csv.reader(f)#注意这里的缩进应该是书的印刷错误

header =reader.next()

data= [row for row inreader]except:

...略...

开始在python3运行报错如下：

Traceback (most recent call last):

File "F:\Techonolgoy\Python\file\blog\csv_open.py", line 8, in

header = reader.next()

AttributeError: '_csv.reader' object has no attribute 'next'

查资料发现在python3里面，reader已经没有了next属性。python3中用力如下：

import csv

with open('stocks.csv') as f:

f_csv = csv.reader(f)

headers = next(f_csv)

for row in f_csv:

# Process row

...

上文提到过，这里书的缩进在印刷上可能有问题(header和data语句应该在with语句块里面)。我按照原缩进，在python2和3两种环境下测试的时候，报同样的错如下：

Traceback (most recent call last):

File "F:\Techonolgoy\Python\file\blog\csv_open.py", line 8, in

header = next(reader)

ValueError: I/O operation on closed file.

于是，再次修改代码，python3版本完整代码如下(python2只需要将缩进改正即可，直接用下面的代码亦可)

import csv

filename = 'ch02-data.csv'

data = []

try:

with open(filename) as f:

reader = csv.reader(f)

header = next(reader)

data = [row for row in reader]

except csv.Error as e:

print("Error reading CSV file at line %s: %s"%(reader.line_num, e))

if header:

print(header)

print("=====================")

for datarow in data:

print(datarow)

成功输出结果：

['day', 'ammount']

=====================

['2013-01-24', '323']

['2013-01-25', '233']

['2013-01-26', '433']

['2013-01-27', '555']

['2013-01-28', '123']

['2013-01-29', '0']

['2013-01-30', '221']