Buscar número de columnas en el archivo csv

Question

Mi programa necesita leer archivos csv que pueden tener 1,2 o 3 columnas, y debe modificar su comportamiento en consecuencia. ¿Hay una manera simple de verificar el número de columnas sin "consumir" una fila antes de que se ejecute el iterador? El siguiente código es el más elegante que pude, pero yo preferiría correr el cheque antes de que el bucle comienza:

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.") 

Editar: debería haber incluido más información sobre mis datos. Si solo hay un campo, debe contener un nombre en notación científica. Si hay dos campos, el primero debe contener un nombre y el segundo un código de enlace. Si hay tres campos, el campo adicional contiene un indicador que especifica si el nombre es actualmente válido. Por lo tanto, si una fila tiene 1, 2 o 3 columnas, todas deben tener lo mismo.

Answer 1

Puede utilizar itertools.tee

itertools.tee (iterable [, n = 2])
Volver n iteradores independientes a partir de una sola iterable.

por ejemplo.

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

Tenga en cuenta que es importante suprimir la referencia a reader1 cuando haya terminado con él - de lo contrario tee tendrá que almacenar todas las filas de la memoria en caso de que alguna vez llama next(reader1) nuevo

Answer 2

¿Qué pasa si el el usuario le proporciona un archivo CSV con menos columnas? ¿Se usan valores predeterminados en su lugar?

En caso afirmativo, ¿por qué no extender la fila con valores nulos?

reader = csv.reader(f,delimiter=d) 
for row in reader: 
    row += [None] * (3 - len(row)) 
    try: 
     foo, bar, baz = row 
    except ValueError: 
     # Too many values to unpack: too many columns in the CSV 
     raise CSVError("Too many columns in input file.") 

Ahora bar y baz al menos será None y el manejador de excepción se hará cargo de todas las filas de más de 3 artículos.

Answer 3

Esto parece funcionar así:

import csv 

datafilename = 'testfile.csv' 
d = '\t' 
f=open(datafilename,'r') 

reader=csv.reader(f,delimiter=d) 
ncol=len(next(reader)) # Read first line and count columns 
f.seek(0)    # go back to beginning of file 
for row in reader: 
    pass #dostuff 

Answer 4

que reconstruiría la siguiente manera (si el archivo no es demasiado grande):

import csv 
f = 'testfile.csv' 
d = '\t' 

reader = list(csv.reader(f,delimiter=d)) 
fields = len(reader[0]) 
for row in reader: 
    if fields == 1: 
     pass 
    elif fields == 2: 
     pass 
    elif fields == 3: 
     pass 
    else: 
     raise CSVError("Too many columns in input file.")