Chcę uzyskać listę nagłówków kolumn z pandas DataFrame. DataFrame będzie pochodzić z danych wejściowych użytkownika, więc nie będę wiedział ile będzie kolumn ani jak będą się nazywać.
Na przykład, jeśli otrzymam DataFrame jak poniżej:
>>> 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
Chciałbym otrzymać listę taką jak ta:
>>> header_list
['y', 'gdp', 'cap']
Możesz uzyskać wartości jako listę, wykonując:
list(my_dataframe.columns.values)
Możesz też po prostu użyć: (jak pokazano w Ed Chum's answer):
list(my_dataframe)