Pyplot: chỉ hiển thị 3 dòng đầu tiên trong chú thích

Question

Tôi đang chạy mô phỏng 200 lần và vẽ 3 danh sách đầu ra dưới dạng 3 dòng có độ trong suốt cao. Điều này cho phép tôi thể hiện sự khác biệt giữa các mô phỏng.

Vấn đề là chú thích của tôi hiển thị 3x200 mục thay vì 3 mục. Tôi làm cách nào để hiển thị chú thích cho từng dòng một lần?

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

thêm

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

cho bất kỳ âm mưu mà bạn không muốn hiển thị trong truyền thuyết. như vậy trong mã của bạn bạn có thể ví dụ làm:

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

và tương tự cho người khác, hoặc nếu bạn không thích mã iffy:

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

Answer 2

Để tránh thêm logic trong âm mưu của mình, sử dụng 'proxy' nghệ sĩ cho các mục nhập huyền thoại của bạn:

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