Comment sélectionner des colonnes par nom dans Pandas (3 exemples)

Vous pouvez utiliser les méthodes suivantes pour sélectionner les colonnes par nom dans un DataFrame pandas :

Méthode 1 : sélectionner une colonne par nom

df.loc[:, 'column1']

Méthode 2 : sélectionner plusieurs colonnes par nom

df.loc[:, ['column1', 'column3', 'column4']] 

Méthode 3 : sélectionner les colonnes dans la plage par nom

df.loc[:, 'column2':'column4'] 

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

import pandas as pd

#create DataFrame
df = pd.DataFrame({'mavs': [10, 12, 14, 15, 19, 22, 27],
                   'cavs': [18, 22, 19, 14, 14, 11, 20],
                   'hornets': [5, 7, 7, 9, 12, 9, 14],
                   'spurs': [10, 12, 14, 13, 13, 19, 22],
                   'nets': [10, 14, 25, 22, 25, 17, 12]})

#view DataFrame
print(df)

   mavs  cavs  hornets  spurs  nets
0    10    18        5     10    10
1    12    22        7     12    14
2    14    19        7     14    25
3    15    14        9     13    22
4    19    14       12     13    25
5    22    11        9     19    17
6    27    20       14     22    12

Exemple 1 : sélectionner une colonne par nom

Le code suivant montre comment sélectionner la colonne « spurs » dans le DataFrame :

#select column with name 'spurs'
df.loc[:, 'spurs']

0    10
1    12
2    14
3    13
4    13
5    19
6    22
Name: spurs, dtype: int64

Seules les valeurs de la colonne « spurs » sont renvoyées.

Exemple 2 : sélectionner plusieurs colonnes par nom

Le code suivant montre comment sélectionner les colonnes cavs, spurs et nets dans le DataFrame :

#select columns with names cavs, spurs, and nets
df.loc[:, ['cavs', 'spurs', 'nets']]

        cavs	spurs	nets
0	18	10	10
1	22	12	14
2	19	14	25
3	14	13	22
4	14	13	25
5	11	19	17
6	20	22	12

Seules les valeurs des colonnes cavs, spurs et nets sont renvoyées.

Exemple 3 : sélectionner les colonnes d’une plage par nom

Le code suivant montre comment sélectionner toutes les colonnes entre les noms « frelons » et « nets » dans le DataFrame :

#select all columns between hornets and nets
df.loc[:, 'hornets':'nets']

        hornets	spurs	nets
0	5	10	10
1	7	12	14
2	7	14	25
3	9	13	22
4	12	13	25
5	9	19	17
6	14	22	12

Toutes les colonnes situées entre les noms « frelons » et « nets » sont renvoyées.

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres tâches courantes dans les pandas :

Pandas : Comment déplacer une colonne devant le DataFrame
Pandas : comment vérifier si la colonne contient une chaîne
Pandas : Comment ajouter une colonne vide à DataFrame (3 exemples)