Obtenir la liste des en-têtes de colonne d'un DataFrame pandas

Je veux obtenir une liste des en-têtes de colonne d'un DataFrame pandas. Le DataFrame proviendra de l'entrée de l'utilisateur, je ne saurai donc pas combien de colonnes il y aura ni comment elles s'appelleront.

Par exemple, si l'on me donne un DataFrame comme celui-ci :

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

je voudrais obtenir une liste comme celle-ci :

>>> header_list
['y', 'gdp', 'cap']
Gulzar
Question modifiée 16 juin 2019 в 11:48
Programmation
python pandas dataframe
20 octobre 2013 в 9:18
21 vues
Solution

Vous pouvez obtenir les valeurs sous forme de liste en faisant :

list(my_dataframe.columns.values)

Vous pouvez également utiliser simplement : (comme indiqué dans la réponse de Ed Chum) :

list(my_dataframe)
cs95
Réponse modifiée 3 avril 2019 в 9:51
Commentaires (10)

Il est disponible sous le nom de my_dataframe.columns.

BrenBarn
Réponse modifiée 23 janvier 2014 в 6:50
Commentaires (2)
 
n = []
for i in my_dataframe.columns:
    n.append(i)
print n
Commentaires (3)