python - 绘图 : show only first 3 lines in legend

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',...