Comment extraire les listes d'en-têtes de colonnes des DataFrames Pandas ?

Comment obtenir une liste d'en-têtes à partir des colonnes Pandas DataFrame

Extraire les en-têtes de colonnes d'un Pandas DataFrame est une tâche courante dans l'analyse des données. Cet article propose deux méthodes pratiques pour obtenir une liste d'en-têtes de colonnes, vous permettant de travailler facilement avec vos données.

Méthode 1 : my_dataframe.columns.values

Cette méthode utilise l'attribut columns de le DataFrame pour accéder à ses en-têtes de colonnes. En appelant .values ​​sur l'objet colonnes renvoyé, vous pouvez convertir les en-têtes en liste. Cela vous donne un contrôle précis sur la liste résultante.

<code class="python">column_headers = list(my_dataframe.columns.values)</code>

Méthode 2 : list(my_dataframe)

Pour une approche plus simple, vous pouvez directement taper le DataFrame lui-même dans une liste. Cette méthode génère également une liste d'en-têtes de colonnes.

<code class="python">column_headers = list(my_dataframe)</code>

Exemple

Considérez le DataFrame suivant :

y	gdp	cap
1	2	5
2	3	9
8	7	2
3	4	7
6	7	7
4	8	3
8	2	8
9	9	10
6	6	4
10	10	7

En utilisant la méthode 1 ou la méthode 2, vous obtenez la liste suivante d'en-têtes de colonnes :

['y', 'gdp', 'cap']

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!