在matplotlib中显示一些图形而不停止计算

import matplotlib.pyplot as plt

# 开启交互模式
plt.ion()

# 创建一个图形窗口
fig, ax = plt.subplots()

# 循环更新图形
for i in range(10):
    # 清空图形
    ax.clear()
    
    # 绘制新的图形
    x = range(i+1)
    y = [n**2 for n in x]
    ax.plot(x, y)
    
    # 更新图形
    plt.draw()
    plt.pause(0.5)  # 暂停一段时间，以便观察图形

# 关闭交互模式
plt.ioff()

# 显示图形
plt.show()