Comment créer une légende unique pour plusieurs sous-parcelles Matplotlib ?

Création d'une légende unique pour plusieurs sous-parcelles Matplotlib

Lorsque vous tracez des informations similaires sur plusieurs sous-parcelles à l'aide de Matplotlib, il peut être avantageux de créer une seule légende. légende qui s’applique à toutes les sous-parcelles. Cela simplifie l'interprétation des données en fournissant une référence cohérente pour les lignes de chaque sous-intrigue.

Pour y parvenir, utilisez la fonction get_legend_handles_labels() sur le dernier axe ou en appelant plt.gca().get_legend_handles_labels( ) si vous utilisez l'interface pyplot. Ces fonctions collectent les poignées et les étiquettes de légende nécessaires à partir des arguments label=.

Pour créer la légende unique, appelez fig.legend(handles, labels, loc='upper center'), où fig est la figure contenant les sous-parcelles et loc spécifient l'emplacement de la légende.

Par exemple, si vous avez une grille 3x3 de sous-parcelles avec des lignes identiques, le code suivant ajouterait une seule légende au-dessus de toutes les sous-parcelles :

import matplotlib.pyplot as plt
import numpy as np

# Generate data for the subplots
data = np.random.rand(9)

# Create the subplots
fig, axes = plt.subplots(3, 3)

# Plot the data on each subplot
for ax, datum in zip(axes.flatten(), data):
    ax.plot(datum)

# Get the legend handles and labels
handles, labels = plt.gca().get_legend_handles_labels()

# Create the single legend
plt.legend(handles, labels, loc='upper center')

plt.show()

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!