You can iterate over your line objects list, so labels are individually assigned. An example with the built-in python iter
function:
lineObjects = plt.plot(x, y)
plt.legend(iter(lineObjects), ('foo', 'bar', 'baz'))`
Edit: after updating to matplotlib 1.1.1, it looks like the plt.plot(x, y)
, with y as a list of lists (as provided by the author of the question), doesn’t work anymore. The one step plotting without iteration over the y arrays is still possible thought after passing y as numpy.array
(assuming (numpy)[http://numpy.scipy.org/] as been previously imported).
In this case, use plt.plot(x, y)
(if the data in the 2D y array are arranged as columns [axis 1]) or plt.plot(x, y.transpose())
(if the data in the 2D y array are arranged as rows [axis 0])
Edit 2: as pointed by @pelson (see commentary below), the iter
function is unnecessary and a simple plt.legend(lineObjects, ('foo', 'bar', 'baz'))
works perfectly