patternpythonCritical
Get a list from Pandas DataFrame column headers
Viewed 0 times
pandaslistgetheaderscolumndataframefrom
Problem
I want to get a list of the column headers from a Pandas DataFrame. The DataFrame will come from user input, so I won't know how many columns there will be or what they will be called.
For example, if I'm given a DataFrame like this:
I would get a list like this:
For example, if I'm given a DataFrame like this:
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
I would get a list like this:
['y', 'gdp', 'cap']
Solution
You can get the values as a list by doing:
Also you can simply use (as shown in Ed Chum's answer):
list(my_dataframe.columns.values)Also you can simply use (as shown in Ed Chum's answer):
list(my_dataframe)Code Snippets
list(my_dataframe.columns.values)list(my_dataframe)Context
Stack Overflow Q#19482970, score: 2071
Revisions (0)
No revisions yet.