Želim dobiti seznam naslovov stolpcev iz podatkovnega okvira pandas DataFrame. DataFrame bo prišel iz uporabniškega vnosa, zato ne bom vedel, koliko stolpcev bo ali kako se bodo imenovali.
Na primer, če dobim DataFrame, kot je ta:
>>> 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
želim dobiti seznam, kot je ta:
>>> header_list
['y', 'gdp', 'cap']
Vrednosti lahko pridobite kot seznam, tako da naredite:
list(my_dataframe.columns.values)
Uporabite lahko tudi preprosto: (kot je prikazano v odgovoru Eda Chuma):
list(my_dataframe)