Pandas : Comment obtenir le nom de la colonne par index

Vous pouvez utiliser les méthodes suivantes pour obtenir un nom de colonne par position d’index dans les pandas :

Méthode 1 : obtenir un nom de colonne par position d’index

#get column name in index position 2
colname = df.columns[2]

Méthode 2 : obtenir plusieurs noms de colonnes par positions d’index

#get column names in index positions 2 and 4
colname = df.columns[[2, 4]] 

Les exemples suivants montrent comment utiliser chaque méthode en pratique avec le DataFrame pandas suivant :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   'points': [18, 22, 19, 14, 14, 11, 20, 28],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4],
                   'rebounds': [11, 8, 10, 6, 6, 5, 9, 12],
                   'steals': [4, 3, 3, 2, 5, 4, 3, 8],
                   'blocks': [1, 0, 0, 3, 2, 2, 1, 5]})

#view DataFrame
print(df)

  team  points  assists  rebounds  steals  blocks
0    A      18        5        11       4       1
1    B      22        7         8       3       0
2    C      19        7        10       3       0
3    D      14        9         6       2       3
4    E      14       12         6       5       2
5    F      11        9         5       4       2
6    G      20        9         9       3       1
7    H      28        4        12       8       5

Exemple 1 : obtenir un nom de colonne par position d’index

Le code suivant montre comment obtenir le nom de la colonne en position d’index 2 du DataFrame :

#get column name for column in index position 2
colname = df.columns[2]

#display column name
print(colname)

assists

Le nom de la colonne en position d’index 2 est assists .

Remarque : les valeurs d’index de colonne commencent à 0 en Python. Ainsi, la colonne en position d’index 2 serait la troisième colonne du DataFrame, qui porte le nom assists .

Exemple 2 : obtenir plusieurs noms de colonnes par positions d’index

Le code suivant montre comment obtenir les noms de colonnes pour les colonnes aux positions d’index 2 et 4 du DataFrame :

#get column names in index positions 2 and 4
colname = df.columns[[2, 4]] 

#display column names
print(colname)

Index(['assists', 'steals'], dtype='object')

À partir du résultat, nous pouvons voir :

• Le nom de la colonne en position d’index 2 est assists .
• Le nom de la colonne en position d’index 4 est Stees .

Remarque : Dans cet exemple, nous avons choisi d’obtenir les noms de colonnes pour deux colonnes par position d’index, mais vous pouvez utiliser une syntaxe similaire pour récupérer les noms de colonnes pour autant de colonnes que vous le souhaitez.

Ressources additionnelles

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

Comment obtenir la valeur d’une cellule à partir de Pandas DataFrame
Comment renommer l’index dans Pandas DataFrame
Comment trier les colonnes par nom dans Pandas