Matplotlib - 2D and 3D plotting in Python
matplotlib - 2D and 3D plotting in Python
J.R. Johansson (robert@riken.jp)
The latest version of this IPython notebook () lecture is available at ().
The other notebooks in this lecture series are indexed at ().
In [1]:
# This line configures matplotlib to show figures embedded in the notebook, # instead of opening a new window for each figure. More about that later. # If you are using an old version of IPython, try using '%pylab inline' instead. %matplotlib inline
Introduction
Matplotlib is an excellent 2D and 3D graphics library for generating scientific figures. Some of the many advantages of this library include: Easy to get started
Support for LATEX formatted labels and texts
Great control of every element in a figure, including figure size and DPI. High-quality output in many formats, including PNG, PDF, SVG, EPS, and PGF. GUI for interactively exploring figures and support for headless generation of figure files (useful for batch jobs). One of the of the key features of matplotlib that I would like to emphasize, and that I think makes matplotlib highly suitable for generating figures for scientific publications is that all aspects of the figure can be controlled programmatically. This is important for reproducibility and convenient when one needs to regenerate the figure with updated data or change its appearance. More information at the Matplotlib web page:
To get started using Matplotlib in a Python program, either include the symbols from the pylab module (the easy way):
In [2]: from pylab import *
or import the matplotlib.pyplot module under the name plt (the tidy way):
In [3]: import matplotlib.pyplot as plt
MATLAB-like API
The easiest way to get started with plotting using matplotlib is often to use the MATLAB-like API provided by matplotlib. It is designed to be compatible with MATLAB's plotting functions, so it is easy to get started with if you are familiar with MATLAB. To use this API from matplotlib, we need to include the symbols in the pylab module:
In [4]: from pylab import *
Example
A simple figure with MATLAB-like plotting API:
In [5]: x = linspace(0, 5, 10) y = x ** 2
In [6]:
figure() plot(x, y, 'r') xlabel('x') ylabel('y') title('title') show()
Most of the plotting related functions in MATLAB are covered by the pylab module. For example, subplot and color/symbol selection:
In [7]:
subplot(1,2,1) plot(x, y, 'r--') subplot(1,2,2) plot(y, x, 'g*-');
The good thing about the pylab MATLAB-style API is that it is easy to get started with if you are familiar with MATLAB, and it has a minumum of coding overhead for simple plots. However, I'd encourrage not using the MATLAB compatible API for anything but the simplest figures. Instead, I recommend learning and using matplotlib's object-oriented plotting API. It is remarkably powerful. For advanced figures with subplots, insets and other components it is very nice to work with.
The matplotlib object-oriented API
The main idea with object-oriented programming is to have objects that one can apply functions and actions on, and no object or program states should be global (such as the MATLAB-like API). The real advantage of this approach becomes apparent when more than one figure is created, or when a figure contains more than one subplot. To use the object-oriented API we start out very much like in the previous example, but instead of creating a new global figure instance we store a reference to the newly created figure instance in the fig variable, and from it we create a new axis instance axes using the add_axes method in the Figure class instance fig:
In [8]: fig = plt.figure() axes = fig.add_axes([0.1, 0.1, 0.8, 0.8]) # left, bottom, width, height (range 0 to 1) axes.plot(x, y, 'r') axes.set_xlabel('x') axes.set_ylabel('y') axes.set_title('title');
Although a little bit more code is involved, the advantage is that we now have full control of where the plot axes are placed, and we can easily add more than one axis to the figure:
In [9]: fig = plt.figure() axes1 = fig.add_axes([0.1, 0.1, 0.8, 0.8]) # main axes axes2 = fig.add_axes([0.2, 0.5, 0.4, 0.3]) # inset axes # main figure axes1.plot(x, y, 'r') axes1.set_xlabel('x') axes1.set_ylabel('y') axes1.set_title('title') # insert axes2.plot(y, x, 'g') axes2.set_xlabel('y') axes2.set_ylabel('x') axes2.set_title('insert title');
If we don't care about being explicit about where our plot axes are placed in the figure canvas, then we can use one of the many axis layout managers in matplotlib. My favorite is subplots, which can be used like this: In [10]: fig, axes = plt.subplots()
axes.plot(x, y, 'r') axes.set_xlabel('x') axes.set_ylabel('y') axes.set_title('title');
In [11]: fig, axes = plt.subplots(nrows=1, ncols=2) for ax in axes: ax.plot(x, y, 'r') ax.set_xlabel('x') ax.set_ylabel('y') ax.set_title('title')
That was easy, but it isn't so pretty with overlapping figure axes and labels, right? We can deal with that by using the fig.tight_layout method, which automatically adjusts the positions of the axes on the figure canvas so that there is no overlapping content: In [12]: fig, axes = plt.subplots(nrows=1, ncols=2)
for ax in axes: ax.plot(x, y, 'r') ax.set_xlabel('x') ax.set_ylabel('y') ax.set_title('title')
fig.tight_layout()
Figure size, aspect ratio and DPI
Matplotlib allows the aspect ratio, DPI and figure size to be specified when the Figure object is created, using the figsize and dpi keyword arguments. figsize is a tuple of the width and height of the figure in inches, and dpi is the dots-per-inch (pixel per inch). To create an 800x400 pixel, 100 dots-per-inch figure, we can do:
................
................
In order to avoid copyright disputes, this page is only a partial summary.
To fulfill the demand for quickly locating and searching documents.
It is intelligent file search solution for home and business.
Related download
- basic plotting with python and matplotlib
- lab 5 introduction to matplotlib
- read map and extract modis aerosol data using python
- cs294 112 deep reinforcement learning plotting and
- color map
- 5 introduction to matplotlib
- matplotlib 2d and 3d plotting in python
- a hands on introduction to using python in the atmospheric
- 1 introduction download the sample codes