Please use matplotlib instead of pylab.