Manipulations CSV avec python

Question

Pour mes besoins, je dois connaître le nombre de lignes dans le fichier (CSV) avant de travailler avec des lignes. J'ai googlé et trouvé que la documentation dit que je devrais créer un itérateur (CSV.reader) pour deux fois (le premier pour compter et le second pour travailler avec des rangées). Est-ce la seule façon ou peut-être il y a une méthode délicate pour faire un tour?

Merci pour vos réponses.

Answer 1

si votre fichier est pas très grand que vous pouvez essayer:

from csv import reader 

def proceed(size): 
    with open(filename) as f: 
     data = list(csv.reader(f)) 
     if len(data) > size: 
      return 
     else: 
      for line in data: 
       #do action 


weights = {'user1': 4, 'user2': 5} 
for k,v in weights.iteritems(): 
    proceed(v) 

Ou comme suggéré par @georgesl dans le cas où vous avez un très gros fichier:

def proceed(size): 
    if sum(1 for row in csv.reader(open(filename))) > size: 
     return 
    else: 
     for line in csv.reader(open(filename)): 
      #do action 

Answer 2

Je ne sais pas d'une manière sans lire le fichier, mais en fonction de l'endroit où vos goulots d'étranglement sont vous pouvez simplement traiter N lignes, et s'il y a plus de les jeter, par exemple:

count = 0 
for line in reader: 
    count += 1 
    if count > N: # Over the limit so stop processing 
     break 
    else: 
     processed_data += process(line) 
else: 
    # This block only runs if the loop completed naturally, i.e. count <= N 
    return processed_data 

Si process(line) est cher, alors votre meilleur pari peut être d'utiliser deux boucles comme décrit dans votre question.