Pyplot: mostra solo le prime 3 righe nella legenda

Question

Sto eseguendo una simulazione 200 volte e tracciando le 3 liste di output come 3 linee con elevata trasparenza. Questo mi permette di mostrare la varianza tra le simulazioni.

Il problema è che la mia leggenda mostra articoli 3x200 invece di 3 articoli. Come faccio a mostrare la legenda per ogni riga una volta?

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

aggiungere

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

per qualsiasi tracciato che non si desidera visualizzare nella leggenda. così nel codice si può ad esempio fare:

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

e allo stesso modo per gli altri, o se non ti piace il codice incerto:

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

Answer 2

Per evitare la logica in più nel vostro tracciato, uso 'delega' artisti per le vostre voci di legenda:

# 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()