Hent liste fra pandas DataFrame-kolonneoverskrifter

Jeg ønsker at få en liste over kolonneoverskrifterne fra et Pandas DataFrame. DataFrame vil komme fra brugerinput, så jeg vil ikke vide, hvor mange kolonner der vil være, eller hvad de vil blive kaldt.

For eksempel, hvis jeg får et DataFrame som dette:

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

vil jeg gerne have en liste som denne:

>>> header_list
['y', 'gdp', 'cap']
Gulzar
Redigeret spørgsmål 2019ør juni 2019 в 11:48
Programmering
python pandas dataframe
2013ør oktober 2013 в 9:18
27 visninger
Løsning

Du kan få værdierne som en liste ved at gøre:

list(my_dataframe.columns.values)

Du kan også blot bruge: (som vist i Ed Chum's svar):

list(my_dataframe)
cs95
Redigeret svar 2019ør april 2019 в 9:51
Kommentarer (10)

Det er tilgængeligt som my_dataframe.columns.

BrenBarn
Redigeret svar 2014ør januar 2014 в 6:50
Kommentarer (2)
 
n = []
for i in my_dataframe.columns:
    n.append(i)
print n
Kommentarer (3)