Добавьте label=
к каждому из ваших вызовов plot()
, а затем вызовите legend(loc='upper left')
.
Рассмотрим этот образец:
import numpy as np
import pylab
x = np.linspace(0, 20, 1000)
y1 = np.sin(x)
y2 = np.cos(x)
pylab.plot(x, y1, '-b', label='sine')
pylab.plot(x, y2, '-r', label='cosine')
pylab.legend(loc='upper left')
pylab.ylim(-1.5, 2.0)
pylab.show()
Немного изменен из этого урока: http://jakevdp.github.io/mpl_tutorial/tutorial_pages/tut1.html