Wie extrahiere ich Spaltenüberschriften aus einem Pandas DataFrame?

Spaltenüberschriften aus einem Pandas DataFrame abrufen

Pandas DataFrames sind vielseitige Datenstrukturen, die eine effiziente Datenbearbeitung und -analyse ermöglichen. Eine häufige Aufgabe besteht darin, Spaltenüberschriften zu extrahieren, was nützlich sein kann, um einen Überblick über die Struktur des DataFrames zu erhalten oder für die weitere Verarbeitung.

Angenommen, Sie haben einen DataFrame aus Benutzereingaben importiert, bei dem die Anzahl und Namen der Spalten unbekannt sind . Um die Spaltenüberschriften als Liste zu extrahieren, können Sie die folgenden Methoden nutzen:

Methode 1: Verwenden von DataFrame.columns.values

<code class="python">column_headers = list(my_dataframe.columns.values)</code>

my_dataframe.columns gibt zurück ein Indexobjekt, das die Spaltenüberschriften enthält. Indem Sie diesen Index mithilfe von Werten in eine Liste konvertieren, erhalten Sie eine Liste von Zeichenfolgen, die die Spaltennamen darstellen.

Methode 2: Verwenden von DataFrame.columns

<code class="python">column_headers = list(my_dataframe)</code>

Dies Methode ist eine Kurzschreibweise für my_dataframe.columns.values, die die Spalten direkt in eine Liste konvertiert.

Beispielverwendung

Bedenken Sie den DataFrame:

    y  gdp  cap
0   1    2    5
1   2    3    9
2   8    7    2
3   3    4    7
4   6    7    7
5   4    8    3
6   8    2    8
7   9    9   10
8   6    6    4
9  10   10    7

Mit beiden Methoden erhalten Sie die folgende Liste von Spaltenüberschriften:

['y', 'gdp', 'cap']

Das obige ist der detaillierte Inhalt vonWie extrahiere ich Spaltenüberschriften aus einem Pandas DataFrame?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!