Animations with Matplotlib

Using the matplotlib library to create some interesting animations.

Rain Simulation with Matplotlib

Animations are an interesting way of demonstrating a phenomenon. We as humans are always enthralled by animated and interactive charts rather than the static ones. Animations make even more sense when depicting time-series data like stock prices over the years, climate change over the past decade, seasonalities and trends since we can then see how a particular parameter behaves with time.

The above image is a simulation of Rain and has been achieved with Matplotlib library which is fondly known as the grandfather of python visualization packages. Matplotlib simulates raindrops on a surface by animating the scale and opacity of 50 scatter points. Today Python boasts of a large number of powerful visualization tools like Plotly, Bokeh, Altair to name a few. These libraries are able to achieve state of the art animations and interactiveness. Nonetheless, the aim of this article is to highlight one aspect of this library which isn’t explored much and that is Animations and we are going to look at some of the ways of doing that.


Overview

Matplotlib is a Python 2D plotting library and also the most popular one. Most of the people start their Data Visualisation journey with Matplotlib. One can generate plots, histograms, power spectra, bar charts, error charts, scatterplots, etc easily with matplotlib. It also integrates seamlessly with libraries like Pandas and Seaborn to create even more sophisticated visualizations.

Some of the nice features of matplotlib are:

  • It is designed like MATLAB hence switching between the two is fairly easy.
  • Comprises of a lot of rendering backends.
  • It can reproduce just about any plots( with a bit of effort).
  • Has been out there for over a decade, therefore, boasts of a huge user base.

However, there are also areas where Matplotlib doesn’t shine out so much and lags behind its powerful counterparts.

  • Matplotlib has an imperative API which is often overly verbose.
  • Sometimes poor stylistic defaults.
  • Poor support for web and interactive graphs.
  • Often slow for large & complicated data.

As for a refresher here is a Matplotlib Cheatsheet from Datacamp which you can go through to brush up your basics.

https://s3.amazonaws.com/assets.datacamp.com/blog_assets/Python_Matplotlib_Cheat_Sheet.pdf


Animations

Matplotlib’s animation base class deals with the animation part. It provides a framework around which the animation functionality is built. There are two main interfaces to achieve that using:

FuncAnimation makes an animation by repeatedly calling a function func.

ArtistAnimation: Animation using a fixed set of Artist objects.

However, out of the two, FuncAnimation is the most convenient one to use. You can read more about them in the documentation since we will only concern ourselves with the FuncAnimation tool.

Requirements

  • Modules including numpy and matplotlib should be installed.
  • To save the animation on your system as mp4 or gif, ffmpeg or imagemagick is required to be installed.

Once ready, we can begin with our first basic animation in the Jupyter Notebooks. The code for this article can be accessed from the associated Github Repository or you can view it on my binder by clicking the image below.

Basic Animation: Moving Sine Wave

Let’s use FuncAnimation to create a basic animation of a sine wave moving across the screen. The source code for the animation has been taken from the Matplotlib Animation tutorial. Let’s first see the output and then we shall break down the code to understand what’s going under the hood.

https://gist.github.com/parulnith/5d82f66f38704e09456fbc33ed5b67ae


  • In lines(7–9), we simply create a figure window with a single axis in the figure. Then we create our empty line object which is essentially the one to be modified in the animation. The line object will be populated with data later.
  • In lines(11–13), we create the init function that will make the animation happen. The init function initializes the data and also sets the axis limits.
  • In lines(14–18), we finally define the animation function which takes in the frame number(i) as the parameter and creates a sine wave(or any other animation) which a shift depending upon the value of i. This function here returns a tuple of the plot objects which have been modified which tells the animation framework what parts of the plot should be animated.
  • In line 20, we create the actual animation object. The blit parameter ensures that only those pieces of the plot are re-drawn which have been changed.

This is the basic intuition behind creating animations in Matplotlib. With a little tweak in the code, interesting visualizations can be created. Let’s have a look at some of them


A Growing Coil

Similarly, there is a nice example of creating shapes at GeeksforGeeks. Let’s now create a moving coil that slowly unwinds, with the help of animation class of matplotlib. The code is quite similar to the sine wave plot with minor adjustments.

https://gist.github.com/parulnith/2a29f424126e1b313259310d8927bccb



Live Updating Graphs

Live updating graphs come in handy when plotting dynamic quantities like stock data, sensor data or any other time-dependent data. We plot a base graph which automatically gets updated as more data is fed into the system. This example has been taken from sentdex. Be sure to visit this youtube channel for some awesome tutorials.

Let’s plot stock prices of a hypothetical company in a month.

https://gist.github.com/parulnith/0c02c6428af9b11aee24d7514ac36627

Now, open the terminal and run the python file. You will get a graph like the one below which automatically updates as follows:


Here interval is 1000 milliseconds or one second.


Animation on a 3D plot

Creating 3D graphs is common but what if we can animate the angle of view of those graphs. The idea is to change the camera view and then use every resulting image to create an animation. There is a nice section dedicated to it at The Python Graph Gallery.

Create a folder called volcano in the same directory as the notebook. All the images will be stored in this folder which will be then used in the animation.

https://gist.github.com/parulnith/a33cdb15b8888ba40e8f91aa40c0f61c

This will create multiple PNG files in the Volcano folder. Now, use ImageMagick to transform them into animation. Open Terminal and navigate to the Volcano folder and enter the following command:

convert -delay 10 Volcano*.png animated_volcano.gif



Animations using Celluloid Module

Celluloid is a Python module that simplifies the process of creating animations in matplotlib. This library creates a matplotlib figure and creates a Camera from it. It then reuses figure and after each frame is created, take a snapshot with the camera. Finally, an animation is created with all the captured frames.

Installation

pip install celluloid

Here are a few examples using the Celluloid module.

Minimal

https://gist.github.com/parulnith/40f26938ab2ca684e81550cfa1974d64


Subplots

https://gist.github.com/parulnith/99289d99473993c7d311c1be4ecf30c6


Legends

https://gist.github.com/parulnith/8e55323704750049f875684b65b37cde



Wrap Up

Animations help to highlight certain features of the visualisation which otherwise cannot be communicated easily with static charts. Having said that it is also important to keep in mind that unnecessary and overuse of visualisations can sometimes complicate things. Every feature in data visualisation should be used judiciously to have the best impact.

1 Comment »

Leave a Reply

Fill in your details below or click an icon to log in:

WordPress.com Logo

You are commenting using your WordPress.com account. Log Out /  Change )

Google photo

You are commenting using your Google account. Log Out /  Change )

Twitter picture

You are commenting using your Twitter account. Log Out /  Change )

Facebook photo

You are commenting using your Facebook account. Log Out /  Change )

Connecting to %s