Plotting data with Python and PyLab