Comment créer et ajouter plusieurs éléments de légende personnalisés dans Matplotlib ?

Gestion automatisée des légendes ou création manuelle de légendes

Lorsque vous travaillez avec matplotlib, il est souvent souhaitable d'ajouter manuellement des éléments de légende constitués d'une couleur et une étiquette. Cependant, l'ajout manuel de plusieurs éléments de légende peut s'avérer fastidieux et entraîner des doublons.

L'approche décrite par l'utilisateur, ax2.legend(self.labels,colorList[:len(self.labels)]), ne semblent fonctionner comme prévu car matplotlib offre une solution plus élégante pour créer manuellement des éléments de légende.

Création d'éléments de légende personnalisés

Comme suggéré par le Legend Guide, nous pouvons utiliser la classe Patches pour définir des éléments de légende personnalisés. Un patch est une zone rectangulaire qui peut être spécifiée avec une couleur et une étiquette. Par exemple :

<code class="python">import matplotlib.patches as mpatches

red_patch = mpatches.Patch(color='red', label='The red data')</code>

Cela crée un patch de légende avec une zone rectangulaire rouge et l'étiquette « Les données rouges ».

Ajout de plusieurs éléments de légende

Pour ajouter plusieurs correctifs à la légende, nous pouvons utiliser la fonction plt.legend et transmettre une liste de correctifs comme argument handles. Par exemple :

<code class="python">blue_patch = mpatches.Patch(color='blue', label='The blue data')

plt.legend(handles=[red_patch, blue_patch])</code>

Cela donnera lieu à une légende avec deux entrées, une pour les données rouges et une pour les données bleues.

En utilisant cette approche, vous pouvez créer et contrôlez le contenu de votre légende, même lorsque vous traitez un grand nombre d'éléments.

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!