我的程序需要读取可能有1,2或3列的csv文件,并且需要相应地修改它的行为。在迭代器运行之前,是否有一个简单的方法来检查列数,而不“消耗”一行?以下代码是我可以管理的最优雅的,但是我宁愿在for循环开始之前运行检查:
import csv
f = 'testfile.csv'
d = '\t'
reader = csv.reader(f,delimiter=d)
for row in reader:
if reader.line_num == 1: fields = len(row)
if len(row) != fields:
raise CSVError("Number of fields should be %s: %s" % (fields,str(row)))
if fields == 1:
pass
elif fields == 2:
pass
elif fields == 3:
pass
else:
raise CSVError("Too many columns in input file.")
编辑:我应该包括更多有关我的数据的信息。如果只有一个字段,它必须包含科学符号的名称。如果有两个字段,那么第一个必须包含一个名字,第二个是链接代码。如果有三个字段,则附加字段包含一个指定该名称当前是否有效的标志。因此,如果任何一行有1,2或3列,则必须相同。