How to pass arguments to animation.FuncAnimation() in Matplotlib?


To pass arguments to animation.FuncAnimation() for a contour plot in Matplotlib in Python, we can take the following steps −

  • Create a random data of 10☓10 dimension.
  • Create a figure and a set of subplots using subplots() method.
  • Make an animation by repeatedly calling a function *func* using FuncAnimation() class
  • To update the contour value in the function, we can define a method animate() that can be used in FuncAnimation() class.
  • To display the figure, use show() method.

Example

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

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

data = np.random.randn(800).reshape(10, 10, 8)
fig, ax = plt.subplots()

def animate(i):
   ax.clear()
   ax.contourf(data[:, :, i])

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

plt.show()

Output

Updated on: 08-Jul-2021

564 Views

Kickstart Your Career

Get certified by completing the course

Get Started
Advertisements