Ignorer les en-têtes lors de la modification d'un fichier csv à l'aide de Python

J'utilise ci-dessous le code référencé pour modifier un csv en utilisant Python. Les fonctions appelées dans le code forment la partie supérieure du code.

Problème: je veux que le code référencé ci-dessous commence à éditer le csv de la 2ème ligne, je veux qu'il exclue la 1ère ligne qui contient des en-têtes. En ce moment, il applique les fonctions sur la 1ère ligne uniquement et ma ligne d'en-tête est modifiée.

in_file = open("tmob_notcleaned.csv", "rb")
reader = csv.reader(in_file)
out_file = open("tmob_cleaned.csv", "wb")
writer = csv.writer(out_file)
row = 1
for row in reader:
    row[13] = handle_color(row[10])[1].replace(" - ","").strip()
    row[10] = handle_color(row[10])[0].replace("-","").replace("(","").replace(")","").strip()
    row[14] = handle_gb(row[10])[1].replace("-","").replace(" ","").replace("GB","").strip()
    row[10] = handle_gb(row[10])[0].strip()
    row[9] = handle_oem(row[10])[1].replace("Blackberry","RIM").replace("TMobile","T-Mobile").strip()
    row[15] = handle_addon(row[10])[1].strip()
    row[10] = handle_addon(row[10])[0].replace(" by","").replace("FREE","").strip()
    writer.writerow(row)
in_file.close()    
out_file.close()

J'ai essayé de résoudre ce problème en initialisant la rowvariable à 1mais cela n'a pas fonctionné.

Veuillez m'aider à résoudre ce problème.

Votre readervariable est un itérable, en bouclant dessus vous récupérez les lignes.

Pour lui faire sauter un élément avant votre boucle, il suffit d'appeler next(reader, None)et d'ignorer la valeur de retour.

Vous pouvez également simplifier un peu votre code; utilisez les fichiers ouverts comme gestionnaires de contexte pour les fermer automatiquement:

with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
   reader = csv.reader(infile)
   next(reader, None)  # skip the headers
   writer = csv.writer(outfile)
   for row in reader:
       # process each row
       writer.writerow(row)

# no need to close, the files are closed automatically when you get to this point.

Si vous vouliez écrire l'en-tête dans le fichier de sortie non traité, c'est aussi facile, passez la sortie de next()à writer.writerow():

headers = next(reader, None)  # returns the headers or `None` if the input is empty
if headers:
    writer.writerow(headers)

Une autre façon de résoudre ce problème consiste à utiliser la classe DictReader, qui «saute» la ligne d'en-tête et l'utilise pour autoriser l'indexation nommée.

Étant donné "foo.csv" comme suit:

FirstColumn,SecondColumn
asdf,1234
qwer,5678

Utilisez DictReader comme ceci:

import csv
with open('foo.csv') as f:
    reader = csv.DictReader(f, delimiter=',')
    for row in reader:
        print(row['FirstColumn'])  # Access by column header instead of column number
        print(row['SecondColumn'])

Faire row=1ne changera rien, car vous écraserez simplement cela avec les résultats de la boucle.

Vous voulez faire next(reader)pour sauter une ligne.