Dynamically add legends to matplotlib plots in python


I have a set of plots in python and want to add legends to each of them separately. I am generating the plots in a for loop and want to add the legends dynamically.

Im getting only the last legend shown. I want all 9 off them to be displayed

for q in range(1,10):
      matplotlib.pylab.plot(s_A_approx, label = q)
matplotlib.pylab.legend(loc = 'upper left')
4/29/2013 12:05:09 AM

Accepted Answer

I can't reproduce your problem. With a few adjustments to your script, what you're expecting is working for me.

import matplotlib.pylab
import numpy as np

for q in range(1,10):
    # create a random, 100 length array
    s_A_approx = np.random.randint(0, 100, 100)
    # note I had to make q a string to avoid an AttributeError when 
    # initializing the legend
    matplotlib.pylab.plot(s_A_approx, marker='.', linestyle='None', label=str(q))

matplotlib.pylab.legend(loc='upper left')

resulting plot

If it helps, here's my matplotlib version:

>>> import matplotlib
>>> matplotlib.__version__
4/29/2013 12:48:35 PM

Licensed under: CC-BY-SA with attribution
Not affiliated with: Stack Overflow