如何使用matplotlib制作2d网格的动画

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

fig, ax = plt.subplots()

def update(frame):
    # 在此处更新网格数据
    # 例如，可以使用numpy生成一个随机的2D网格
    grid = np.random.rand(10, 10)
    
    # 清除当前子图内容
    ax.clear()
    
    # 绘制2D网格
    ax.imshow(grid, cmap='viridis')
    
    # 设置坐标轴范围
    ax.set_xlim(0, 10)
    ax.set_ylim(0, 10)
    
    # 可选：添加标题、标签等
    ax.set_title('2D Grid Animation')
    ax.set_xlabel('X')
    ax.set_ylabel('Y')

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

plt.show()