Pandas : Comment modifier les noms de colonnes dans un tableau croisé dynamique

Souvent, vous souhaiterez peut-être modifier ou formater les noms de colonnes dans un tableau croisé dynamique pandas d’une manière spécifique.

Heureusement, cela est facile à faire en utilisant les fonctions intégrées aux pandas.

L’exemple suivant montre comment procéder.

Exemple : modifier les noms de colonnes dans le tableau croisé dynamique Pandas

Supposons que nous ayons le DataFrame pandas suivant qui contient des informations sur divers joueurs de basket-ball :

import pandas as pd

#create DataFrame
df = pd.DataFrame({'team': ['A', 'A', 'A', 'A', 'B', 'B', 'B', 'B'],
                   'position': ['G', 'G', 'F', 'C', 'G', 'C', 'F', 'F'],
                   'points': [4, 4, 6, 8, 9, 5, 5, 12]})

#view DataFrame
print(df)

	team	position points
0	A	G	 4
1	A	G	 4
2	A	F	 6
3	A	C	 8
4	B	G	 9
5	B	C	 5
6	B	F	 5
7	B	F	 12

Nous pouvons utiliser le code suivant pour créer un tableau croisé dynamique dans pandas qui montre la valeur moyenne des points pour chaque équipe et position dans le DataFrame :

#create pivot table
piv = pd.pivot_table(df, values='points', index='team', columns='position')

#view pivot table
print(piv)

position    C    F    G
team                   
A         8.0  6.0  4.0
B         5.0  8.5  9.0

Supposons maintenant que nous souhaitions supprimer la position du mot dans le tableau croisé dynamique et supprimer la ligne d’équipe supplémentaire du tableau croisé dynamique.

Nous pouvons utiliser la syntaxe suivante pour ce faire :

#format column names
piv.columns = ['_'.join(str(s).strip() for s in col if s) for col in piv.columns]

#reset index
piv.reset_index(inplace=True)

#view updated pivot table
print(piv)

  team    C    F    G
0    A  8.0  6.0  4.0
1    B  5.0  8.5  9.0

Notez que nous avons pu supprimer la position du mot dans le tableau croisé dynamique et supprimer la ligne d’équipe supplémentaire du tableau croisé dynamique.

Notez également que cette solution générale fonctionnera également pour un tableau croisé dynamique avec un MultiIndex.

Remarque : Vous pouvez trouver la documentation complète de la fonction pandas pivot_table() ici .

Ressources additionnelles

Les didacticiels suivants expliquent comment effectuer d’autres opérations courantes dans les pandas :

Pandas : Comment créer un tableau croisé dynamique avec le nombre de valeurs
Pandas : comment remplacer les valeurs NaN dans le tableau croisé dynamique par des zéros
Pandas : Comment convertir un tableau croisé dynamique en DataFrame