Pyplot：僅顯示圖例中的前3行

Question

我正在運行模擬200次並將3個輸出列表繪製爲3行高透明度。這允許我顯示模擬之間的差異。

問題是我的圖例顯示3x200項目，而不是3項。我怎樣才能讓它爲每一行顯示圖例？

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_') 

爲您不想在圖例中顯示的任何繪圖。所以在你的代碼例如可以做：

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

同樣地，對於他人的，或者如果你不喜歡玄乎代碼：

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

Answer 2

爲避免你的繪圖，使用「代理」額外的邏輯您的傳奇作品的藝術家：

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