我正在尝试在matplotlib中制作轮廓图动画。

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

# 创建x和y的坐标网格
x = np.linspace(-5, 5, 100)
y = np.linspace(-5, 5, 100)
X, Y = np.meshgrid(x, y)

# 创建z的值，这里以二维高斯函数为例
Z = np.exp(-(X**2 + Y**2) / 10) * np.cos(X) * np.sin(Y)

fig, ax = plt.subplots()
contour = ax.contour(X, Y, Z)

def update(frame):
    ax.cla()  # 清除轴上的内容
    # 更新z的值
    Z_new = np.exp(-((X - frame/10)**2 + Y**2) / 10) * np.cos(X) * np.sin(Y)
    # 绘制轮廓图
    contour = ax.contour(X, Y, Z_new)
    return contour

ani = animation.FuncAnimation(fig, update, frames=range(100), interval=100)

plt.show()