Wie schließe ich Header aus, wenn ich CSV-Dateien mit Python verarbeite?

Header bei der Verarbeitung von CSV-Dateien mit Python ausschließen

Bei der Verarbeitung einer CSV-Datei kann es erforderlich sein, Header von bestimmten Vorgängen auszuschließen. Dies kann durch die Verwendung der CSV-Reader- und Writer-Module von Python erreicht werden.

Im bereitgestellten Code ist das Überspringen der Kopfzeile erwünscht. Anstatt die Zeilenvariable auf 1 zu initialisieren, besteht ein einfacherer Ansatz darin, die erste Zeile zu überspringen, bevor der Rest verarbeitet wird. Dies kann wie folgt erreicht werden:

<code class="python">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)</code>

Durch den Aufruf von next(reader, None) wird die erste Zeile der CSV-Datei abgerufen und verworfen. Die folgenden Zeilen können dann verarbeitet und in die Ausgabedatei geschrieben werden.

Darüber hinaus kann der Code vereinfacht werden, indem Kontextmanager verwendet werden, um das Öffnen und Schließen von Dateien automatisch zu handhaben:

<code class="python">with open("tmob_notcleaned.csv", "rb") as infile, open("tmob_cleaned.csv", "wb") as outfile:
    reader = csv.reader(infile)
    writer = csv.writer(outfile)
    # Skip the headers
    if headers := next(reader, None):
        writer.writerow(headers)</code>

Hier Code empfängt die optionale Headers-Variable die übersprungenen Header, sodass der Benutzer sie bei Bedarf unverarbeitet in die Ausgabedatei schreiben kann.

Das obige ist der detaillierte Inhalt vonWie schließe ich Header aus, wenn ich CSV-Dateien mit Python verarbeite?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!