Comment créer une légende unifiée pour plusieurs sous-parcelles dans Matplotlib ?

Création d'une légende unifiée pour plusieurs sous-parcelles dans Matplotlib

Dans Matplotlib, lors du traçage de données sur plusieurs sous-parcelles, il peut souvent être souhaitable d'afficher une seule légende pour toutes les parcelles . Ceci est particulièrement utile lorsque les sous-tracés partagent des styles de ligne, des couleurs ou des étiquettes similaires. En consolidant les légendes, cela permet de simplifier la visualisation et de réduire l'encombrement.

Pour y parvenir, Matplotlib fournit une fonction pratique appelée get_legend_handles_labels(). Cette fonction peut être appelée sur le dernier axe de la grille et collecte toutes les informations de légende nécessaires à partir des arguments label= utilisés lors du traçage. Les poignées et étiquettes collectées peuvent ensuite être utilisées pour créer une légende unique pour la figure entière.

Par exemple, si vous avez neuf sous-parcelles disposées dans une grille 3x3, vous pouvez obtenir les éléments de légende de la dernière sous-parcelle en utilisant le code suivant :

handles, labels = ax.get_legend_handles_labels()

où ax représente le dernier axe de sous-intrigue dans la grille. Avec les poignées et les étiquettes collectées, la légende peut être créée à l'aide de fig.legend() :

fig.legend(handles, labels, loc='upper center')

Dans ce cas, la légende sera positionnée à l'emplacement « supérieur au centre » de la figure. L'argument loc= peut être personnalisé pour placer la légende n'importe où dans la figure.

Alternativement, si vous utilisez l'interface pyplot au lieu de l'interface Axes, le code suivant peut être utilisé pour récupérer les poignées et les étiquettes de légende :

handles, labels = plt.gca().get_legend_handles_labels()

où plt.gca() renvoie l'objet axe actuel.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!