python读取csv某一列并统计行数_python – 查找csv文件中的列数

我的程序需要读取可能有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列，则必须相同。