#!/usr/bin/env python
#
# Make a legend for specific lines.
from pylab import *
t1 = arange(0.0, 2.0, 0.1)
t2 = arange(0.0, 2.0, 0.01)
# note that plot returns a list of lines. The "l1, = plot" usage
# extracts the first element of the list inot l1 using tuple
# unpacking. So l1 is a Line2D instance, not a sequence of lines
l1, = plot(t2, exp(-t2))
l2, l3 = plot(t2, sin(2*pi*t2), '--go', t1, log(1+t1), '.')
l4, = plot(t2, exp(-t2)*sin(2*pi*t2), 'rs-.')
legend( (l2, l4), ('oscillatory', 'damped'), 'upper right', shadow=True)
xlabel('time')
ylabel('volts')
title('Damped oscillation')
#axis([0,2,-1,1])
show()
|