Ik wil een lijst van de kolomkoppen van een pandas DataFrame krijgen. De DataFrame zal afkomstig zijn van input van de gebruiker, dus ik weet niet hoeveel kolommen er zullen zijn of hoe ze genoemd zullen worden.
Bijvoorbeeld, als ik'een DataFrame als dit krijg:
>>> 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
zou ik een lijst als deze willen krijgen:
>>> header_list
['y', 'gdp', 'cap']
U kunt de waarden als een lijst krijgen door te doen:
list(my_dataframe.columns.values)
Je kunt ook gewoon: gebruiken (zoals in Ed Chum's antwoord):
list(my_dataframe)