在for循环绘图时如何清除图形但保留轴标签与标题？

你遇到的问题是plt.cla()会清空整个坐标轴的所有内容，包括标签和标题。下面给你两种实用的解决思路：

方法1：直接删除轴上的线条对象 #

不用清空整个坐标轴，而是只删掉已经绘制的曲线，标签和标题会完整保留。

场景1：每次循环只画一条曲线 #

可以直接删除最新绘制的线条，代码修改如下：

# 先把轴标签、标题一次性设置好（放在循环外面！）
ax.set_xlabel('X轴标签')
ax.set_ylabel('Y轴标签')
ax.set_title('图标题')

for n in range(N):
    ax.plot(x[n], t)  # 绘制当前迭代的曲线
    plt.savefig(f'fig_{n}.png')
    ax.lines.pop()  # 删除刚画的这条线，保留标签标题

场景2：每次循环画多条曲线 #

如果一次画多条，就遍历删除轴上所有线条：

ax.set_xlabel('X轴标签')
ax.set_ylabel('Y轴标签')
ax.set_title('图标题')

for n in range(N):
    # 这里假设一次画多条线
    ax.plot(x1[n], t)
    ax.plot(x2[n], t)
    plt.savefig(f'fig_{n}.png')
    # 删除所有线条
    for line in ax.lines:
        line.remove()

方法2：更新已有线条的数据（更高效） #

如果每次循环只需要更新一条曲线的内容，可以先创建一个空线条，之后在循环里直接更新它的x、y数据，全程不用清除任何内容，标签标题一直保留：

# 先创建一个空的线条对象
line, = ax.plot([], [])
# 一次性设置标签和标题
ax.set_xlabel('X轴标签')
ax.set_ylabel('Y轴标签')
ax.set_title('图标题')
# 提前固定坐标轴范围（如果数据范围固定的话，避免每次自动缩放）
ax.set_xlim(min(x.flatten()), max(x.flatten()))
ax.set_ylim(min(t), max(t))

for n in range(N):
    line.set_data(x[n], t)  # 直接更新线条的数据源
    plt.savefig(f'fig_{n}.png')

注意事项 #

• 标签和标题一定要放在循环外设置，不要重复执行，浪费资源。
• 如果你的数据范围不固定，方法1或者方法2里去掉set_xlim/set_ylim即可，matplotlib会自动调整坐标轴范围。

内容的提问来源于stack exchange，提问作者johnny rotten