Comment faire de la coutume légende dans matplotlib

Je génèrent actuellement ma légende avec matplotlib de cette façon:

if t==25:
    l1,l2 = ax2.plot(x320,vTemp320,'or',x320,vAnaTemp320,'-r')
elif t==50:
    l3,l4 = ax2.plot(x320,vTemp320,'ob',x320,vAnaTemp320,'-b')
else:
    l5,l6 = ax2.plot(x320,vTemp320,'og',x320,vAnaTemp320,'-g')
plt.legend((l1,l2,l3,l4,l5,l6), ('t=25 Simulation', 't=25 Analytical','t=50 Simulation', 't=50 Analytical','t=500 Simulation', 't=500 Analytical'),
   bbox_to_anchor=(-.25, 1), loc=2, borderaxespad=0.,prop={'size':12})

Qui fonctionne en quelque sorte voir Un. Mais j'ai la duplication des informations dans ma légende.

Je préfère séparer la légende. De sorte que j'ai de couleurs différentes lignes correspondant à l'instant t. Et une ligne normale que ma solution Analytique d'un des points pour les résultats de ma simulation.

Quelque chose comme ça

--(ligne rouge) t = 25

--(ligne bleue) t = 50

--(ligne verte) t = 500

o Simulaton

-- Solution Analytique

Quelqu'un maintenant, comment je pourrais réaliser avec matplotlib?

Comment faire de la coutume légende dans matplotlib

InformationsquelleAutor tune2fs | 2012-11-09
  1. 82

    Vous pouvez choisir les artistes et les étiquettes à afficher dans la légende comme suit. Vous aurez besoin de créer des artistes pour les éléments de la légende qui ne sont pas réellement de tracé.

    import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0,10,31)
fig = plt.figure()
ax = fig.add_subplot(1,1,1)
#Plot analytic solution
ax.plot(x,1*x**2, color='r', label="t = 25")
ax.plot(x,2*x**2, color='b', label="t = 50")
ax.plot(x,3*x**2, color='g', label="t = 500")
#Plot simulation
ax.plot(x,1*x**2, color='r', linestyle='', marker='o')
ax.plot(x,2*x**2, color='b', linestyle='', marker='o')
ax.plot(x,3*x**2, color='g', linestyle='', marker='o')
#Get artists and labels for legend and chose which ones to display
handles, labels = ax.get_legend_handles_labels()
display = (0,1,2)
#Create custom artists
simArtist = plt.Line2D((0,1),(0,0), color='k', marker='o', linestyle='')
anyArtist = plt.Line2D((0,1),(0,0), color='k')
#Create legend from custom artist/label lists
ax.legend([handle for i,handle in enumerate(handles) if i in display]+[simArtist,anyArtist],
[label for i,label in enumerate(labels) if i in display]+['Simulation', 'Analytic'])
plt.show()

    Comment faire de la coutume légende dans matplotlib

    • Merci ça fonctionne parfaitement
    • Cela peut être une question de la langue anglaise, mais qu'entendez-vous par "artiste"?
    • Je veux dire le matplotlib artiste de classe
    • Module artist, classe Artist.
    InformationsquelleAutor Mr. Squig