Animation with contours in matplotlib

To animate with contours in matplotlib, we can take the following steps


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

  • Create data for the contour plot.

  • Create a figure and a set of subplots.

  • Generate an animation by repeatedly calling a function *animate* where the animate() method changes the contour data points.

  • To display the figure, use Show() method.


import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation

# Set the figure size
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True

# Random data for the contour plot
data = np.random.randn(800).reshape(10, 10, 8)

# Create a figure and a set of subplots
fig, ax = plt.subplots()

# Method to change the contour data points
def animate(i):
    ax.contourf(data[:, :, i], cmap='plasma')

# Call animate method
ani = animation.FuncAnimation(fig, animate, 5, interval=50, blit=False)

# Display the plot


It will produce the following output −