Wie extrahiere ich mit dem CSV-Modul Daten aus bestimmten Spalten in einer CSV-Datei?

Extrahieren von Daten aus bestimmten Spalten in einer CSV-Datei mit dem CSV-Modul

Bei der Arbeit mit CSV-Dateien ist es oft notwendig, Daten zu extrahieren Nur aus bestimmten Spalten. Um dies mit dem CSV-Modul zu erreichen, ist es wichtig zu verstehen, wie die Bibliothek den Datenabruf handhabt.

Im bereitgestellten Codebeispiel erfolgt der Versuch, auf bestimmte Spalten zuzugreifen, über die row[i]-Syntax, wobei i das darstellt Spaltennummer. Allerdings ist das Ergebnis aufgrund eines falschen Ansatzes nicht wie erwartet.

Die richtige Methode besteht darin, die print-Anweisung in die for-Schleife einzubinden. Der folgende korrigierte Code demonstriert dies:

import csv

included_cols = [1, 2, 6, 7]
with open("csv_file.csv") as csvfile:
    reader = csv.reader(csvfile, delimiter=",")
    for row in reader:
        content = list(row[i] for i in included_cols)
        print(content)  # Include the print statement within the loop to display each row

Jetzt gibt der geänderte Code die Werte aus den gewünschten Spalten für jede Zeile aus.

Es ist jedoch erwähnenswert, dass ein alternativer Ansatz verwendet wird Pandas-Bibliothek, die den Umgang mit CSV-Dateien vereinfacht. Mit Pandas können Sie ganz einfach eine CSV-Datei lesen und bestimmte Spalten in Variablen extrahieren:

import pandas as pd

df = pd.read_csv("csv_file.csv")
names = df["Name"]  # Use df['column_name'] to extract a specific column into a variable

Die Pandas-Bibliothek bietet einen vielseitigen Satz an Tools für die Arbeit mit strukturierten Daten und ist daher eine empfehlenswerte Wahl für den Umgang mit CSV-Dateien effektiv.

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