Récupération des en-têtes de colonne à partir d'un DataFrame Pandas

Les DataFrames Pandas sont des structures de données polyvalentes qui permettent une manipulation et une analyse efficaces des données. Une tâche courante consiste à extraire les en-têtes de colonnes, ce qui peut être utile pour obtenir un aperçu de la structure du DataFrame ou pour un traitement ultérieur.

Supposons que vous ayez un DataFrame importé à partir d'une entrée utilisateur, où le nombre et les noms des colonnes sont inconnus. . Pour extraire les en-têtes de colonnes sous forme de liste, vous pouvez utiliser les méthodes suivantes :

Méthode 1 : Utilisation de DataFrame.columns.values

column_headers = list(my_dataframe.columns.values)

my_dataframe.columns renvoie un objet Index contenant les en-têtes de colonnes. En convertissant cet index en liste utilisant des valeurs, vous obtenez une liste de chaînes représentant les noms de colonnes.

Méthode 2 : Utilisation de DataFrame.columns

column_headers = list(my_dataframe)

Cette méthode est une notation abrégée pour my_dataframe.columns.values, qui convertit directement le colonnes dans une liste.

Exemple d'utilisation

Considérez le 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

En utilisant l'une ou l'autre méthode, vous obtiendrez la liste suivante d'en-têtes de colonnes :

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