Wie extrahiere ich mit Python bestimmte Spalten aus einer CSV-Datei?

CSV-Spalten mit dem Python-CSV-Modul kürzen

Oft ist es notwendig, eine CSV-Datei (Comma-Separated Values) zu analysieren und Extrahieren Sie bestimmte Datenspalten. Mit dem CSV-Modul von Python kann man diese Aufgabe problemlos erledigen.

Spaltenextraktion

Um bestimmte Spalten zu extrahieren, verwenden Sie die Variable „included_cols“, die die zu verwendenden Spaltenindizes angibt enthalten. Jede Spalte ist beginnend mit 0 nummeriert, wobei die erste Spalte 0 ist. Um beispielsweise die Spalten „ID“, „Name“, „Postleitzahl“ und „Telefon“ zu erfassen, würden Sie include_cols = [1, 2, 6 schreiben , 7].

Iteration und Drucken

Um die Zeilen zu durchlaufen und die angegebenen Spalten zu extrahieren, verwenden Sie ein Listenverständnis innerhalb einer for-Schleife. Für jede Zeile wird die Inhaltsvariable mit den Werten aus den angegebenen Spalten gefüllt. Um die extrahierten Daten anzuzeigen, verwenden Sie print(content) innerhalb der Schleife.

included_cols = [1, 2, 6, 7]

for row in reader:
    content = list(row[i] for i in included_cols)
    print(content)

Optimieren mit Pandas

Für eine effizientere CSV-Verarbeitung sollten Sie die Verwendung des Pandas-Moduls in Betracht ziehen . Pandas bietet praktische und leistungsstarke Tools zum Lesen und Bearbeiten von CSV-Dateien. Um eine CSV-Datei zu lesen und eine bestimmte Spalte in einer Variablen zu speichern, verwenden Sie einfach:

import pandas as pd
df = pd.read_csv(csv_file)
saved_column = df.column_name

Das obige ist der detaillierte Inhalt vonWie extrahiere ich mit Python bestimmte Spalten aus einer CSV-Datei?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!