Liste von Pandas DataFrame Spaltenüberschriften abrufen

Ich möchte eine Liste der Spaltenüberschriften aus einem Pandas DataFrame zu erhalten. Der DataFrame wird von Benutzereingaben kommen, so dass ich nicht weiß, wie viele Spalten es sein wird oder wie sie genannt werden.

Zum Beispiel, wenn ich ein DataFrame wie diese gegeben werden:

>>> my_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

Ich würde eine Liste wie diese erhalten wollen:

>>> header_list
['y', 'gdp', 'cap']
Lösung

Sie können die Werte als Liste erhalten, indem Sie dies tun:

list(my_dataframe.columns.values)

Sie können auch einfach: verwenden (wie in Ed Chums Antwort gezeigt):

list(my_dataframe)
Kommentare (10)

Diese ist als "my_dataframe.columns" verfügbar.

Kommentare (2)
n = []
for i in my_dataframe.columns:
    n.append(i)
print n
Kommentare (3)