Trouver le nombre de colonnes dans le fichier csv

Question

Mon programme a besoin de lire les fichiers csv qui peuvent avoir 1,2 ou 3 colonnes, et il doit modifier son comportement en conséquence. Existe-t-il un moyen simple de vérifier le nombre de colonnes sans "consommer" une ligne avant l'exécution de l'itérateur? Le code suivant est le plus élégant que je pouvais gérer, mais je préférerais courir le chèque avant que la boucle commence:

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

Edit: j'ai inclus plus d'informations sur mes données. S'il n'y a qu'un seul champ, il doit contenir un nom en notation scientifique. S'il y a deux champs, le premier doit contenir un nom et le second un code de liaison. S'il existe trois champs, le champ supplémentaire contient un indicateur qui spécifie si le nom est actuellement valide. Donc si une ligne a 1, 2 ou 3 colonnes, toutes doivent avoir la même chose.

Answer 1

Vous pouvez utiliser itertools.tee

itertools.tee (itérables [, n = 2])
Retour n itérateurs indépendants à partir d'un seul itérables.

par ex.

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

Notez qu'il est important de supprimer la référence à reader1 lorsque vous avez terminé avec elle - sinon tee devra stocker toutes les lignes en mémoire au cas où vous appelez next(reader1) à nouveau

Answer 2

Que se passe si la l'utilisateur vous fournit un fichier CSV avec moins de colonnes? Les valeurs par défaut sont-elles utilisées à la place?

Si oui, pourquoi ne pas étendre la ligne avec des valeurs nulles à la place?

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

maintenant bar et baz sera au moins None et le gestionnaire d'exception se chargera de toutes les lignes de plus de 3 articles.

Answer 3

Cela semble fonctionner aussi bien:

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

Je rebâtirait comme suit (si le fichier est pas trop grand):

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