Get list from pandas DataFrame column headers

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:

    >>> 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

I would like to get a list like the one below:

    >>> header_list
    [y, gdp, cap]

You can get the values as a list by doing:

    list(my_dataframe.columns.values)

Also you can simply use:

    list(my_dataframe)

From: stackoverflow.com/q/19482970

Back to homepage or read more recommendations: