Pandas: so verwenden sie read_csv mit dem argument usecols

Sie können das Argument usecols in der Funktion read_csv() verwenden, um bestimmte Spalten aus einer CSV-Datei in einen Pandas-DataFrame zu lesen.

Es gibt zwei gängige Möglichkeiten, dieses Argument zu verwenden:

Methode 1: Verwenden Sie usecols mit Spaltennamen

 df = pd. read_csv (' my_data.csv ', usecols=[' this_column ', ' that_column '])

Methode 2: Usecols mit Spaltenpositionen verwenden

 df = pd. read_csv ( ' my_data.csv ', usecols=[ 0,2 ])

Die folgenden Beispiele zeigen, wie jede Methode in der Praxis mit der folgenden CSV-Datei namens „basketball_data.csv“ verwendet wird:

Beispiel 1: Verwendung von Usecols mit Spaltennamen

Wir können den folgenden Code verwenden, um die CSV-Datei zu importieren und nur die Spalten „Team“ und „Rebounds“ zu verwenden:

 import pandas as pd

#import DataFrame and only use 'team' and 'rebounds' columns
df = pd. read_csv (' basketball_data.csv ', usecols=[' team ',' rebounds '])

#view DataFrame
print (df)

   team rebounds
0 to 10
1 B 9
2 C 6
3 D 2

Beachten Sie, dass nur die Spalten „Team“ und „Rebounds“ importiert wurden, da dies die Spaltennamen waren, die wir im Argument „usecols“ angegeben haben.

Beispiel 2: Verwendung von Usecols mit Spaltenpositionen

Wir können den folgenden Code verwenden, um die CSV-Datei zu importieren und nur die Spalten an den Indexpositionen 0 und 2 zu verwenden:

 import pandas as pd

#import DataFrame and only use columns in index positions 0 and 2
df = pd. read_csv (' basketball_data.csv ' , usecols=[ 0,2 ])

#view DataFrame
print (df)

   team rebounds
0 to 10
1 B 9
2 C 6
3 D 2

Beachten Sie, dass nur die Spalten „Team“ und „Rebounds“ importiert wurden, da es sich hierbei um die Spalten an den Indexpositionen 0 und 2 handelte, also den Werten, die wir im Argument „usecols“ angegeben haben.

Hinweis : Die erste Spalte der CSV-Datei hat die Indexposition 0.

Zusätzliche Ressourcen

Die folgenden Tutorials erklären, wie Sie andere häufige Aufgaben in Python ausführen:

Pandas: So überspringen Sie Zeilen beim Lesen einer CSV-Datei
Pandas: So lesen Sie Excel-Dateien
Pandas: So exportieren Sie einen DataFrame nach Excel