Pyplot: zeige nur die ersten 3 Zeilen in der Legende

Question

Ich führe eine Simulation 200 Mal und zeichne die 3 Ausgabelisten als 3 Zeilen mit hoher Transparenz. Dies erlaubt mir, Unterschiede zwischen den Simulationen zu zeigen.

Das Problem ist, dass meine Legende 3x200 Elemente statt 3 Elemente zeigt. Wie bekomme ich es, die Legende für jede Zeile einmal anzuzeigen?

for simulation in range(200): 
    plt.plot(num_s_nodes, label="susceptible", color="blue", alpha=0.02) 
    plt.plot(num_r_nodes, label="recovered", color="green", alpha=0.02) 
    plt.plot(num_i_nodes, label="infected", color="red", alpha=0.02) 
plt.legend() 
plt.show() 

Answer 1

plt.plot(... , label='_nolegend_') 

für jede Plotten hinzufügen, die Sie nicht wollen, in der Legende zu zeigen. so in Ihrem Code können Sie zum Beispiel tun:

..., label='_nolegend_' if simulation else 'susceptible', ... 

und in ähnlicher Weise für andere, oder wenn Sie nicht strittig Code mögen:

..., label=simulation and '_nolegend_' or 'susceptible',... 

Answer 2

Um zusätzliche Logik in Ihren Plotten zu vermeiden, verwenden Sie ‚Proxy‘ Künstler für Ihre Legende Einträge:

# no show lines for you ledgend 
plt.plot([], label="susceptible", color="blue", alpha=0.02) 
plt.plot([], label="recovered", color="green", alpha=0.02) 
plt.plot([], label="infected", color="red", alpha=0.02) 

for simulation in range(200): 
    # your actual lines 
    plt.plot(num_s_nodes, color="blue", alpha=0.02) 
    plt.plot(num_r_nodes, color="green", alpha=0.02) 
    plt.plot(num_i_nodes, color="red", alpha=0.02) 
plt.legend() 
plt.show()