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']
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)