如何使用matplotlib实时绘制更新的numpy ndarray？

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

fig, ax = plt.subplots()

data = np.zeros((10, 10))

def update_plot(frame):
    # 更新数据
    data = np.random.rand(10, 10)
    
    # 清空子图内容
    ax.clear()
    
    # 绘制新的数据
    ax.imshow(data, cmap='hot')
    
    # 设置标题和标签等
    ax.set_title('Real-time Plot')
    ax.set_xlabel('X-axis')
    ax.set_ylabel('Y-axis')

ani = animation.FuncAnimation(fig, update_plot, interval=100)

plt.show()