Quero obter uma lista dos cabeçalhos de coluna de um DataFrame pandas. O DataFrame virá da entrada do usuário, então eu ganhei'não sei quantas colunas haverá ou como elas serão chamadas.
Por exemplo, se I'm der um DataFrame como este:
>>> 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
Eu gostaria de ter uma lista como esta:
>>> header_list
['y', 'gdp', 'cap']
Você pode obter os valores como uma lista ao fazer:
list(my_dataframe.columns.values)
Você também pode simplesmente usar: (como mostrado em Ed Chum's answer):
list(my_dataframe)