matplotlib动画中的函数参数是如何定义的？

Question

大家好，我是 python/matplotlib 的新手。

我很难理解 matplotlib 网站上的以下动画代码。如果在 中调用此函数，def update 中的 data 参数如何定义animation.FuncAnimation 没有指定任何 data 作为输入参数？

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

fig, ax = plt.subplots()
line, = ax.plot(np.random.rand(10))
ax.set_ylim(0, 1)


def update(data):
    line.set_ydata(data)
    return line,


def data_gen():
    while True:
        yield np.random.rand(10)

ani = animation.FuncAnimation(fig, update, data_gen, interval=100)
plt.show()

Answer 1

data_gen 是一个生成器，它为每个动画帧生成一个包含 10 个随机值的数组。从data_gen 返回的数组是传递给update 函数的第一个输入参数。

如果您使用fargs kwarg 调用FuncAnimation，您还可以将其他 输入传递给update 函数。

def update(data, scale):
    line.set_ydata(scale * data)
    return line

animation.FuncAnimation(fig, update, data_gen, fargs=(100,), interval=100)