Comment modifier l’ordre des éléments dans la légende Matplotlib

Vous pouvez utiliser le morceau de code suivant pour modifier l’ordre des éléments dans une légende Matplotlib :

#get handles and labels
handles, labels = plt.gca().get_legend_handles_labels()

#specify order of items in legend
order = [1,2,0]

#add legend to plot
plt.legend([handles[idx] for idx in order],[labels[idx] for idx in order])

L’exemple suivant montre comment utiliser cette syntaxe dans la pratique.

Exemple : modifier l’ordre des éléments dans la légende Matplotlib

Supposons que nous créions le graphique linéaire suivant dans Matplotlib :

import pandas as pd
import matplotlib.pyplot as plt

#create data
df = pd.DataFrame({'points': [11, 17, 16, 18, 22, 25, 26, 24, 29],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4, 8],
                   'rebounds': [6, 8, 8, 10, 14, 12, 12, 10, 11]})

#add lines to plot
plt.plot(df['points'], label='Points', color='green')
plt.plot(df['assists'], label='Assists', color='steelblue')
plt.plot(df['rebounds'], label='Rebounds', color='purple')

#add legend
plt.legend()

Les éléments de la légende sont placés dans l’ordre dans lequel nous avons ajouté les lignes au tracé.

Cependant, nous pouvons utiliser la syntaxe suivante pour personnaliser l’ordre des éléments dans la légende :

import pandas as pd
import matplotlib.pyplot as plt

#create data
df = pd.DataFrame({'points': [11, 17, 16, 18, 22, 25, 26, 24, 29],
                   'assists': [5, 7, 7, 9, 12, 9, 9, 4, 8],
                   'rebounds': [6, 8, 8, 10, 14, 12, 12, 10, 11]})

#add lines to plot
plt.plot(df['points'], label='Points', color='green')
plt.plot(df['assists'], label='Assists', color='steelblue')
plt.plot(df['rebounds'], label='Rebounds', color='purple')

#get handles and labels
handles, labels = plt.gca().get_legend_handles_labels()

#specify order of items in legend
order = [1,2,0]

#add legend to plot
plt.legend([handles[idx] for idx in order],[labels[idx] for idx in order]) 

Notez que nous avons précisé :

• ordre = [1, 2, 0]

Cela signifie:

• Le premier élément de la légende doit être l’étiquette qui se trouvait à l’origine en position d’index 1 de l’ancienne légende (« Aides »).
• Le deuxième élément de la légende doit être l’étiquette qui se trouvait à l’origine en position d’index 2 de l’ancienne légende (« Rebonds »)
• Le troisième élément de la légende doit être l’étiquette qui se trouvait à l’origine en position d’index 0 de l’ancienne légende (« Points »)

Ressources additionnelles

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

Comment changer la position d’une légende dans Matplotlib
Comment placer la légende en dehors d’un tracé Matplotlib
Comment changer la taille de la police de la légende dans Matplotlib