At this point we
would like to introduce you to another resource, which we will be using to generate graphs for the examples – the matplotlib
libraries. It may be downloaded from its official web page, matplotlib.org, and installed following the usual Python motions. There is a good online documentation in the official web page, and we encourage the reader to dig deeper than the few commands that we will use in this book. For instance, the excellent monograph Matplotlib for Python Developers, Sandro Tosi, Packt Publishing, provides all we shall need and more. Other plotting libraries are available (commercial or otherwise), which aim to very different and specific applications. The degree of sophistication and ease of use of matplotlib
makes it one of the best options for generation of graphics in scientific computing.
Once installed, it may be imported as usual, with import matplotlib
. Among all its modules, we will focus on pyplot
, which provides a comfortable interface with the plotting libraries. For example, if we desire to plot at this point a cycle of the sine function, we could execute the following code snippet:
import numpy import matplotlib.pyplot x=numpy.linspace(0,numpy.pi,32) fig=matplotlib.pyplot.figure() fig.plot(x, numpy.sin(x)) fig.savefig('sine.png')
We obtain the following plot:
Let us explain each command from the previous session. The first two commands are used to import numpy
and matplotlib.pyplot
as usual. We define an array x
of 32 uniformly spaced floating point values from 0 to π, and define y
to be the array containing the sine of the values from x
. The command figure creates space in memory to store the subsequent plots, and puts in place an object of the form matplotlib.figure.Figure
. The command plot(x, numpy.sin(x))
creates an object of the form matplotlib.lines.Line2D
, containing data with the plot of x
against numpy.sin(x)
, together with a set of axes attached to it, labeled according to the ranges of the variables. This object is stored in the previous Figure
object. The last command in the session, savefig
, saves the Figure
object to whatever valid image format we desire (in this case, a Portable Network Graphics [PNG] image). If instead of saving to a file we desire to show on screen the result of the plot, we issue the fig.show()
command. From now on, in any code that deals with matplotlib
commands, we will leave the option of showing/saving open.
There are, of course, commands that control the style of axes, aspect ratio between axes, labeling, colors, the possibility of managing several figures at the same time (subplots), and many more options to display all sort of data. We will be discovering these as we progress with examples through the book.