Show legend_unit.py syntax highlighted
from pylab import figure, show, nx
Ntests = 3
t = nx.arange(0.0, 1.0, 0.05)
s = nx.sin(2*nx.pi*t)
# scatter creates a RegPolyCollection
fig = figure()
ax = fig.add_subplot(Ntests, 1, 1)
N = 100
x, y = 0.9*nx.mlab.rand(2,N)
area = nx.pi*(10 * nx.mlab.rand(N))**2 # 0 to 10 point radiuses
ax.scatter(x,y,s=area, marker='^', c='r', label='scatter')
ax.legend()
# vlines creates a LineCollection
ax = fig.add_subplot(Ntests, 1, 2)
ax.vlines(t, [0], nx.sin(2*nx.pi*t), label='vlines')
ax.legend()
# vlines creates a LineCollection
ax = fig.add_subplot(Ntests, 1, 3)
ax.plot(t, s, 'b-', lw=2, label='a line')
ax.legend()
fig.savefig('legend_unit')
show()
See more files for this project here