Ottenere un elenco dalle intestazioni di colonna del DataFrame di pandas

Voglio ottenere un elenco delle intestazioni di colonna da un DataFrame pandas. Il DataFrame verrà dall'input dell'utente quindi non saprò quante colonne ci saranno o come saranno chiamate.

Per esempio, se mi viene dato un DataFrame come questo:

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

Vorrei ottenere una lista come questa:

>>> header_list
['y', 'gdp', 'cap']
Gulzar
Domanda modificata 16º giugno 2019 в 11:48
Programmazione
python pandas dataframe
20º ottobre 2013 в 9:18
18 visualizzazioni
Soluzione

Potete ottenere i valori come lista facendo:

list(my_dataframe.columns.values)

Potete anche usare semplicemente: (come mostrato nella risposta di Ed Chum's):

list(my_dataframe)
cs95
Risposta modificata 3º aprile 2019 в 9:51
Commentari (10)

Questo è disponibile come my_dataframe.columns.

BrenBarn
Risposta modificata 23º gennaio 2014 в 6:50
Commentari (2)
 
n = []
for i in my_dataframe.columns:
    n.append(i)
print n
Commentari (3)