在 Python matplotlib 中添加圖例

Question

我使用 for 循環生成了帶有多行的 plot。 但我無法在該圖中添加圖例。 下面是我嘗試過的代碼。

for l in range(3,8,1):
     seq=np.arange(0,20)
     X=stats.poisson(l)
     seq_pmf = X.pmf(seq)
     plt.plot(seq, seq_pmf,label=l)

我希望圖例標簽為 λ=3, λ=4 ......同樣明智

Answer 1

您需要在顯示圖表之前添加plt.legend() 。

例如：

for l in range(3,8,1):
     seq=np.arange(0,20)
     X=stats.poisson(l)
     seq_pmf = X.pmf(seq)
     plt.plot(seq, seq_pmf,label=l)
plt.legend()
plt.show()

Answer 2

像這樣？

for l in range(3,8,1):
    seq=np.arange(0,20)
    X=stats.poisson(l)
    seq_pmf = X.pmf(seq)
    plt.plot(seq, seq_pmf,label=r'$\lambda$ = '+str(l))
    plt.legend()