Überspringen Sie die Header, wenn Sie eine CSV-Datei mit Python bearbeiten

Ich verwende den unten angegebenen Code, um eine CSV mit Python zu bearbeiten. Im Code aufgerufene Funktionen bilden den oberen Teil des Codes.

Problem: Ich möchte, dass der unten angegebene Code die Bearbeitung der CSV aus der 2. Zeile startet. Ich möchte, dass die 1. Zeile, die Überschriften enthält, ausgeschlossen wird. Im Moment werden die Funktionen nur in der 1. Zeile angewendet und meine Kopfzeile wird geändert.

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()

Ich habe versucht, dieses Problem durch Initialisieren der rowVariablen auf zu lösen , 1aber es hat nicht funktioniert.

Bitte helfen Sie mir bei der Lösung dieses Problems.

Ihre readerVariable ist iterierbar. Wenn Sie sie durchlaufen, rufen Sie die Zeilen ab.

Um ein Element vor Ihrer Schleife zu überspringen, rufen Sie einfach next(reader, None)den Rückgabewert auf und ignorieren Sie ihn.

Sie können Ihren Code auch ein wenig vereinfachen. Verwenden Sie die geöffneten Dateien als Kontextmanager, damit sie automatisch geschlossen werden:

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.

Wenn Sie den Header unverarbeitet in die Ausgabedatei schreiben möchten, ist dies ebenfalls einfach. Übergeben Sie die Ausgabe von next()an writer.writerow():

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

Eine andere Möglichkeit, dies zu lösen, besteht darin, die DictReader-Klasse zu verwenden, die die Kopfzeile "überspringt" und damit die benannte Indizierung zulässt.

Geben Sie "foo.csv" wie folgt an:

FirstColumn,SecondColumn
asdf,1234
qwer,5678

Verwenden Sie DictReader wie folgt:

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'])

Wenn row=1Sie dies tun , ändert sich nichts, da Sie dies nur mit den Ergebnissen der Schleife überschreiben.

Sie möchten next(reader)eine Zeile überspringen.