import numpy as np import matplotlib.pyplot as plt import matplotlib as mpl
A Top-Down runnable Jupyter Notebook with the exact contents of this blog can be found here
An interactive version of this guide can be accessed on Google Colab
A word before we get started…#
Although a beginner can follow along with this guide, it is primarily meant for people who have at least a basic knowledge of how Matplotlib’s plotting functionality works.
Essentially, if you know how to take 2 NumPy arrays and plot them (using an appropriate type of graph) on 2 different axes in a single figure and give it basic styling, you’re good to go for the purposes of this guide.
If you feel you need some introduction to basic Matplotlib plotting, here’s a great guide that can help you get a feel for introductory plotting using Matplotlib
From here on, I will be assuming that you have gained sufficient knowledge to follow along this guide.
Also, in order to save everyone’s time, I will keep my explanations short, terse and very much to the point, and sometimes leave it for the reader to interpret things (because that’s what I’ve done throughout this guide for myself anyway).
The primary driver in this whole exercise will be code and not text, and I encourage you to spin up a Jupyter notebook and type in and try out everything yourself to make the best use of this resource.
What this guide is and what it is not:#
This is not a guide about how to beautifully plot different kinds of data using Matplotlib, the internet is more than full of such tutorials by people who can explain it way better than I can.
This article attempts to explain the workings of some of the foundations of any plot you create using Matplotlib. We will mostly refrain from focusing on what data we are plotting and instead focus on the anatomy of our plots.
Matplotlib has many styles available, we can see the available options using:
['seaborn-dark', 'seaborn-darkgrid', 'seaborn-ticks', 'fivethirtyeight', 'seaborn-whitegrid', 'classic', '_classic_test', 'fast', 'seaborn-talk', 'seaborn-dark-palette', 'seaborn-bright', 'seaborn-pastel', 'grayscale', 'seaborn-notebook', 'ggplot', 'seaborn-colorblind', 'seaborn-muted', 'seaborn', 'Solarize_Light2', 'seaborn-paper', 'bmh', 'tableau-colorblind10', 'seaborn-white', 'dark_background', 'seaborn-poster', 'seaborn-deep']
We shall use
seaborn. This is done like so:
Let’s get started!
# Creating some fake data for plotting xs = np.linspace(0, 2 * np.pi, 400) ys = np.sin(xs**2) xc = np.linspace(0, 2 * np.pi, 600) yc = np.cos(xc**2)
The usual way to create a plot using Matplotlib goes somewhat like this:
fig, ax = plt.subplots(2, 2, figsize=(16, 8)) # `Fig` is short for Figure. `ax` is short for Axes. ax[0, 0].plot(xs, ys) ax[1, 1].plot(xs, ys) ax[0, 1].plot(xc, yc) ax[1, 0].plot(xc, yc) fig.suptitle("Basic plotting using Matplotlib") plt.show()
Our goal today is to take apart the previous snippet of code and understand all of the underlying building blocks well enough so that we can use them separately and in a much more powerful way.
If you’re a beginner like I was before writing this guide, let me assure you: this is all very simple stuff.
plt.subplots documentation (hit
Shift+Tab+Tab in a Jupyter notebook) reveals some of the other Matplotlib internals that it uses in order to give us the
Figure and it’s
These include :
Let’s try and figure out what these functions / classes do.
What is a
Figure? And what are
Figure in Matplotlib is simply your main (imaginary) canvas. This is where you will be doing all your plotting / drawing / putting images and what not. This is the central object with which you will always be interacting. A figure has a size defined for it at the time of creation.
You can define a figure like so (both statements are equivalent):
fig = mpl.figure.Figure(figsize=(10, 10)) # OR fig = plt.figure(figsize=(10, 10))
Notice the word imaginary above. What this means is that a Figure by itself does not have any place for you to plot. You need to attach/add an
Axes to it to do any kind of plotting. You can put as many
Axes objects as you want inside of any
Figure you have created.
- Has a space (like a blank Page) where you can draw/plot data.
- A parent
- Has properties stating where it will be placed inside it’s parent
- Has methods to draw/plot different kinds of data in different ways and add custom styles.
You can create an
Axes like so (both statements are equivalent):
ax1 = mpl.axes.Axes(fig=fig, rect=[0, 0, 0.8, 0.8], facecolor="red") # OR ax1 = plt.Axes(fig=fig, rect=[0, 0, 0.8, 0.8], facecolor="red") #
The first parameter
fig is simply a pointer to the parent
Figure to which an Axes will belong.
The second parameter
rect has four numbers :
[left_position, bottom_position, height, width] to define the position of the
Axes inside the
Figure and the height and width with respect to the
Figure. All these numbers are expressed in percentages.
Figure simply holds a given number of
Axes at any point of time
We will go into some of these design decisions in a few moments'
plt.subplots with basic Matplotlib functionality#
We will try and recreate the below plot using Matplotlib primitives as a way to understand them better. We’ll try and be a slightly creative by deviating a bit though.
fig, ax = plt.subplots(2, 2) fig.suptitle("2x2 Grid")
Text(0.5, 0.98, '2x2 Grid')
Let’s create our first plot using Matplotlib primitives:#
# We first need a figure, an imaginary canvas to put things on fig = plt.Figure(figsize=(6, 6)) # Let's start with two Axes with an arbitrary position and size ax1 = plt.Axes(fig=fig, rect=[0.3, 0.3, 0.4, 0.4], facecolor="red") ax2 = plt.Axes(fig=fig, rect=[0, 0, 1, 1], facecolor="blue")
Now you need to add the
fig. You should stop right here and think about why would there be a need to do this when
fig is already a parent of
ax2? Let’s do this anyway and we’ll go into the details afterwards.
<matplotlib.axes._axes.Axes at 0x1211dead0>
# As you can see the Axes are exactly where we specified. fig
That means you can do this now:
Remark: Notice the
ax.reverse()call in the snippet below. If I hadn’t done that, the biggest plot would be placed in the end on top of every other plot and you would just see a single, blank ‘cyan’ colored plot.
fig = plt.figure(figsize=(6, 6)) ax =  sizes = np.linspace(0.02, 1, 50) for i in range(50): color = str(hex(int(sizes[i] * 255)))[2:] if len(color) == 1: color = "0" + color color = "#99" + 2 * color ax.append(plt.Axes(fig=fig, rect=[0, 0, sizes[i], sizes[i]], facecolor=color)) ax.reverse() for axes in ax: fig.add_axes(axes) plt.show()
The above example demonstrates why it is important to decouple the process of creation of an
Axes and actually putting it onto a
Also, you can remove an
Axes from the canvas area of a
Figure like so:
This can be useful when you want to compare the same primary data (GDP) to several secondary data sources (education, spending, etc.) one by one (you’ll need to add and delete each graph from the Figure in succession)
I also encourage you to look into the documentation for
Axes and glance over the several methods available to them. This will help you know what parts of the wheel you do not need to rebuild when you’re working with these objects the next time.
Recreating our subplots literally from scratch#
This should now make sense. We can now create our original
plt.subplots(2, 2) example using the knowledge we have thus gained so far.
(Although, this is definitely not the most convenient way to do this)
fig = mpl.figure.Figure() fig fig.suptitle("Recreating plt.subplots(2, 2)") ax1 = mpl.axes.Axes(fig=fig, rect=[0, 0, 0.42, 0.42]) ax2 = mpl.axes.Axes(fig=fig, rect=[0, 0.5, 0.42, 0.42]) ax3 = mpl.axes.Axes(fig=fig, rect=[0.5, 0, 0.42, 0.42]) ax4 = mpl.axes.Axes(fig=fig, rect=[0.5, 0.5, 0.42, 0.42]) fig.add_axes(ax1) fig.add_axes(ax2) fig.add_axes(ax3) fig.add_axes(ax4) fig
GridSpec objects allow us more intuitive control over how our plot is exactly divided into subplots and what the size of each
You can essentially decide a Grid which all your
Axes will conform to when laying themselves over.
Once you define a grid, or
GridSpec so to say, you can use that object to generate new
Axes conforming to the grid which you can then add to your
Lets see how all of this works in code:
You can define a
GridSpec object like so (both statements are equivalent):
gs = mpl.gridspec.GridSpec(nrows, ncols, width_ratios, height_ratios) # OR gs = plt.GridSpec(nrows, ncols, width_ratios, height_ratios)
gs = plt.GridSpec(nrows=3, ncols=3, width_ratios=[1, 2, 3], height_ratios[3, 2, 1])
ncols are pretty self explanatory.
width_ratios determines the relative width of each column.
height_ratios follows along the same lines.
grid will always distribute itself using all the space available to it inside of a figure (things change up a bit when you have multiple
GridSpec objects for a single figure, but that’s for you to explore!). And inside of a
grid, all the Axes will conform to the sizes and ratios defined already
def annotate_axes(fig): """Taken from https://matplotlib.org/gallery/userdemo/demo_gridspec03.html#sphx-glr-gallery-userdemo-demo-gridspec03-py takes a figure and puts an 'axN' label in the center of each Axes """ for i, ax in enumerate(fig.axes): ax.text(0.5, 0.5, "ax%d" % (i + 1), va="center", ha="center") ax.tick_params(labelbottom=False, labelleft=False)
fig = plt.figure() # We will try and vary axis sizes here just to see what happens gs = mpl.gridspec.GridSpec(nrows=2, ncols=2, width_ratios=[1, 2], height_ratios=[4, 1])
<Figure size 576x396 with 0 Axes>
You can pass
GridSpec objects to a
Figure to create subplots in your desired sizes and proportions like so :
Notice how the sizes of the
Axes relates to the ratios we defined when creating the Grid.
fig.clear() ax1, ax2, ax3, ax4 = [ fig.add_subplot(gs), fig.add_subplot(gs), fig.add_subplot(gs), fig.add_subplot(gs), ] annotate_axes(fig) fig
Doing the same thing in a simpler way
def add_gs_to_fig(fig, gs): "Adds all `SubplotSpec`s in `gs` to `fig`" for g in gs: fig.add_subplot(g)
fig.clear() add_gs_to_fig(fig, gs) annotate_axes(fig) fig
That means you can now do this:
(Notice how the
Axes sizes increase from top-left to bottom-right)
fig = plt.figure(figsize=(14, 10)) length = 6 gs = plt.GridSpec( nrows=length, ncols=length, width_ratios=list(range(1, length + 1)), height_ratios=list(range(1, length + 1)), ) add_gs_to_fig(fig, gs) annotate_axes(fig) for ax in fig.axes: ax.plot(xs, ys) plt.show()
A very unexpected observation: (which gives us yet more clarity, and Power)#
Notice how after each print operation, different addresses get printed for each
gs, gs, gs, gs
(<matplotlib.gridspec.SubplotSpec at 0x1282a9e50>, <matplotlib.gridspec.SubplotSpec at 0x12942add0>, <matplotlib.gridspec.SubplotSpec at 0x12942a750>, <matplotlib.gridspec.SubplotSpec at 0x12a727e10>)
gs, gs, gs, gs
(<matplotlib.gridspec.SubplotSpec at 0x127d5c6d0>, <matplotlib.gridspec.SubplotSpec at 0x12b6d0b10>, <matplotlib.gridspec.SubplotSpec at 0x129fc6390>, <matplotlib.gridspec.SubplotSpec at 0x129fc6a50>)
print(gs[0, 0], gs[0, 1], gs[1, 0], gs[1, 1])
<matplotlib.gridspec.SubplotSpec object at 0x12951a610> <matplotlib.gridspec.SubplotSpec object at 0x12951a890> <matplotlib.gridspec.SubplotSpec object at 0x12951ac10> <matplotlib.gridspec.SubplotSpec object at 0x12951a150>
print(gs[0, 0], gs[0, 1], gs[1, 0], gs[1, 1])
<matplotlib.gridspec.SubplotSpec object at 0x128fad4d0> <matplotlib.gridspec.SubplotSpec object at 0x1291ebbd0> <matplotlib.gridspec.SubplotSpec object at 0x1294f9850> <matplotlib.gridspec.SubplotSpec object at 0x128106250>
Lets understand why this happens:
Notice how a group of
gs objects indexed into at the same time also produces just one object instead of multiple objects
gs[:, :], gs[:, 0] # both output just one object each
(<matplotlib.gridspec.SubplotSpec at 0x128116e50>, <matplotlib.gridspec.SubplotSpec at 0x128299290>)
# Lets try another `gs` object, this time a little more crowded # I chose the ratios randomly gs = mpl.gridspec.GridSpec( nrows=3, ncols=3, width_ratios=[1, 2, 1], height_ratios=[4, 1, 3] )
All these operations print just one object. What is going on here?
print(gs[:, 0]) print(gs[1:, :2]) print(gs[:, :])
<matplotlib.gridspec.SubplotSpec object at 0x12a075fd0> <matplotlib.gridspec.SubplotSpec object at 0x128cf0990> <matplotlib.gridspec.SubplotSpec object at 0x12a075fd0>
Let’s try and add subplots to our
see what’s going on.
We’ll do a few different permutations to get an exact idea.
fig = plt.figure(figsize=(5, 5)) ax1 = fig.add_subplot(gs[:2, 0]) ax2 = fig.add_subplot(gs[2, 0]) ax3 = fig.add_subplot(gs[:, 1:]) annotate_axes(fig)
fig = plt.figure(figsize=(5, 5)) # ax1 = fig.add_subplot(gs[:2, 0]) ax2 = fig.add_subplot(gs[2, 0]) ax3 = fig.add_subplot(gs[:, 1:]) annotate_axes(fig)
fig = plt.figure(figsize=(5, 5)) # ax1 = fig.add_subplot(gs[:2, 0]) # ax2 = fig.add_subplot(gs[2, 0]) ax3 = fig.add_subplot(gs[:, 1:]) annotate_axes(fig)
fig = plt.figure(figsize=(5, 5)) # ax1 = fig.add_subplot(gs[:2, 0]) # ax2 = fig.add_subplot(gs[2, 0]) ax3 = fig.add_subplot(gs[:, 1:]) # Notice the line below : You can overlay Axes using `GridSpec` too ax4 = fig.add_subplot(gs[2:, 1:]) ax4.set_facecolor("orange") annotate_axes(fig)
fig.clear() add_gs_to_fig(fig, gs) annotate_axes(fig) fig
Here’s a bullet point summary of what this means:
gscan be used as a sort of a
factoryfor different kinds of
- You give this
factoryan order by indexing into particular areas of the
Grid. It gives back a single
type(gs) object that helps you create an
Axeswhich has all of the area you indexed into combined into one unit.
widthratios for the indexed portion will determine the size of the
Axesthat gets generated.
Axeswill maintain relative proportions according to your
- For all these reasons, I like
This ability to create different grid variations that
GridSpec provides is probably the reason for that anomaly we saw a while ago (printing different Addresses).
It creates new objects every time you index into it because it will be very troublesome to store all permutations of
SubplotSpec objects into one group in memory (try and count permutations for a
GridSpec of 10x10 and you’ll know why)
Now let’s finally create
plt.subplots(2,2) once again using GridSpec#
fig = plt.figure() gs = mpl.gridspec.GridSpec(nrows=2, ncols=2) add_gs_to_fig(fig, gs) annotate_axes(fig) fig.suptitle("We're done!") print("yayy")
What you should try:#
Here’s a few things I think you should go ahead and explore:
GridSpecobjects for the Same Figure.
- Deleting and adding
Axeseffectively and meaningfully.
- All the methods available for
mpl.axes.Axesallowing us to manipulate their properties.
- Kaggle Learn’s Data visualization course is a great place to learn effective plotting using Python
- Armed with knowledge, you will be able to use other plotting libraries such as
altairwith much more flexibility (you can pass an
Axesobject to all their plotting functions). I encourage you to explore these libraries too.
This is the first time I’ve written any technical guide for the internet, it may not be as clean as tutorials generally are. But, I’m open to all the constructive criticism that you may have for me (drop me an email on firstname.lastname@example.org)