Enable FFmpeg for Matplotlib Animation

To enable ffmpeg for matplotlib.animation, we can take the following steps −

• Set the figure size and adjust the padding between and around the subplots.

• Set the ffmpeg directory.

• Create a new figure or activate an existing figure, using figure() method.

• Add an 'ax1' to the figure as part of a subplot arrangement.

• Plot the divider based on the pre-existing axes.

• Create random data to be plotted, to display the data as an image, i.e., on a 2D regular raster.

• Create a colorbar for a ScalarMappable instance, cb.

• Set the title as the current frame.

• Make a list of colormaps.

• Make an animation by repeatedly calling a function, animate. The function creates new random data, then use imshow() method to display the data as an image.

• Get an instance of Pipe-based ffmpeg writer.

• Save the current animated figure.

Example

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from mpl_toolkits.axes_grid1 import make_axes_locatable

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True
plt.rcParams['animation.ffmpeg_path'] = 'ffmpeg'

fig = plt.figure()
ax = fig.add_subplot(111)
div = make_axes_locatable(ax)
cax = div.append_axes('right', '5%', '5%')
data = np.random.rand(5, 5)
im = ax.imshow(data)
cb = fig.colorbar(im, cax=cax)
tx = ax.set_title('Frame 0')

cmap = ["copper", 'RdBu_r', 'Oranges', 'cividis', 'hot', 'plasma']

def animate(i):
   cax.cla()
   data = np.random.rand(5, 5)
   im = ax.imshow(data, cmap=cmap[i%len(cmap)])
   fig.colorbar(im, cax=cax)
   tx.set_text('Frame {0}'.format(i))

ani = animation.FuncAnimation(fig, animate, frames=10)
FFwriter = animation.FFMpegWriter()
ani.save('plot.mp4', writer=FFwriter)

Output

When we execute the code, it will create an mp4 file with the name 'plot.mp4' and save it in the Project Directory.