Hämta en lista från pandas DataFrame kolumnrubriker

Jag vill hämta en lista över kolumnrubrikerna från en pandas DataFrame. DataFrame kommer att komma från användarinmatning så jag vet inte hur många kolumner det kommer att finnas eller vad de kommer att kallas.

Om jag till exempel får en DataFrame som denna:

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

Jag skulle vilja få en lista som denna:

>>> header_list
['y', 'gdp', 'cap']
Gulzar
Redigerad fråga 2019ör juni 2019 в 11:48
Programmering
python pandas dataframe
2013ör oktober 2013 в 9:18
24 visningar
Lösning

Du kan få värdena som en lista genom att göra:

list(my_dataframe.columns.values)

Du kan också helt enkelt använda: (som visas i Ed Chum's svar):

list(my_dataframe)
cs95
Redigerat svar 2019ör april 2019 в 9:51
Kommentarer (10)

Det finns som my_dataframe.columns.

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