Comment répertorier tous les noms de colonnes dans Pandas (4 méthodes)

Vous pouvez utiliser l’une des quatre méthodes suivantes pour répertorier tous les noms de colonnes d’un DataFrame pandas :

Méthode 1 : utiliser des crochets

[column for column in df]

Méthode 2 : utilisez tolist()

df.columns.values.tolist()

Méthode 3 : utiliser list()

list(df)

Méthode 4 : utilisez list() avec les valeurs de colonne

list(df.columns.values)

Les exemples suivants montrent comment utiliser chacune de ces méthodes avec le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19, 23],
                   'assists': [5, 7, 7, 9, 12, 9],
                   'rebounds': [11, 8, 10, 6, 6, 5],
                   'blocks': [6, 6, 3, 2, 7, 9]})

#view DataFrame
df

	points	assists	rebounds blocks
0	25	5	11	 6
1	12	7	8	 6
2	15	7	10	 3
3	14	9	6	 2
4	19	12	6	 7
5	23	9	5	 9

Méthode 1 : utiliser des crochets

Le code suivant montre comment répertorier tous les noms de colonnes d’un DataFrame pandas à l’aide de crochets :

[column for column in df]

['points', 'assists', 'rebounds', 'blocks']

Méthode 2 : utilisez tolist()

Le code suivant montre comment lister tous les noms de colonnes à l’aide de la fonction .tolist() :

df.columns.values.tolist()

['points', 'assists', 'rebounds', 'blocks'] 

Méthode 3 : utiliser list()

Le code suivant montre comment lister tous les noms de colonnes à l’aide de la fonction list() :

list(df)

['points', 'assists', 'rebounds', 'blocks'] 

Méthode 4 : utilisez list() avec les valeurs de colonne

Le code suivant montre comment lister tous les noms de colonnes à l’aide de la fonction list() avec les valeurs de colonne :

list(df.columns.values)

['points', 'assists', 'rebounds', 'blocks'] 

Notez que les quatre méthodes renvoient les mêmes résultats.

Notez que pour les DataFrames extrêmement volumineux, la méthode df.columns.values.tolist() a tendance à être la plus rapide.

Ressources additionnelles

Les didacticiels suivants expliquent comment exécuter d’autres fonctions courantes avec les colonnes d’un DataFrame pandas :

Comment supprimer des colonnes dans Pandas
Comment exclure des colonnes dans Pandas
Comment appliquer une fonction aux colonnes sélectionnées dans Pandas
Comment modifier l’ordre des colonnes dans Pandas DataFrame