python - 查找csv文件中的列数

Question

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

Answer 1

您可以使用 itertools.tee

itertools.tee(iterable[, n=2])
Return n independent iterators from a single iterable.

例如。

reader1, reader2 = itertools.tee(csv.reader(f, delimiter=d))
columns = len(next(reader1))
del reader1
for row in reader2:
    ...

请注意，在完成后删除对 reader1 的引用很重要 - 否则 tee 将不得不将所有行存储在内存中，以防你调用next(reader1) 再次