Animation with Contours in Matplotlib

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

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.

Example

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.clear()
    ax.contourf(data[:, :, i], cmap='plasma')

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

# Display the plot
plt.show()

Output

It will produce the following output −