Adding legends in Python matplotlib

Question

I have generated a plot with multiple line using a for loop. But I cannot add a legend to this graph. Below is the code I have tried.

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)

I want the legend labels as λ=3, λ=4...like wise

Yaron Grushka · Accepted Answer · 2020-06-05 12:59:32Z

1

You need to add plt.legend() before showing the graph.

e.g:

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

answered Jun 5, 2020 at 12:59

Yaron Grushka

2332 silver badges12 bronze badges

Sign up to request clarification or add additional context in comments.

3 Comments

user11607046 Over a year ago

Yes. But how can I get it as, λ=3, λ=4...like wise. Instead of just the λ value?

Yaron Grushka Over a year ago

In the label argument you can specify label="$\lambda$=l".

Yaron Grushka Over a year ago

Oh, right. You can use f-string: label=f'$\lambda$ = {l}'

Alexey · Accepted Answer · 2020-06-05 13:08:25Z

1

like this?

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

edited Jun 5, 2020 at 13:08

answered Jun 5, 2020 at 13:00

Alexey

1,1268 silver badges8 bronze badges

2 Comments

user11607046 Over a year ago

plt.plot(seq, seq_pmf,label="λ={0}".format(l)) This works!

Alexey Over a year ago

right. it almost the same. The difference only is how you create this final label string: explicitly use greek lambda-symbol or it code

Collectives™ on Stack Overflow

Adding legends in Python matplotlib

2 Answers 2

3 Comments

2 Comments

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

3 Comments

2 Comments

Related