Matplotlib绘图问题：如何在已有画布添加虚线而非新建画布

问题出在第一次调用plt.show()时，Matplotlib默认会关闭并释放当前绘图窗口的画布资源，所以后续的plt.axhline()会自动创建新画布，plt.clf()是清空当前画布，对这个场景完全没用。

下面给两种可行的解决方法：

方法一：开启交互式模式 #

通过开启Matplotlib的交互式模式，让绘图窗口显示后不阻塞程序执行，后续可以直接更新画布：

import matplotlib.pyplot as plt

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

# 绘制初始曲线
plt.plot([1, 2, 3, 4], [1, 4, 9, 16], 'b-')
# 显示窗口但不阻塞程序
plt.show(block=False)

# 获取用户输入的y值
y_value = float(input("Enter the y-value for the line: "))

# 在已有画布上添加红色虚线
plt.axhline(y=y_value, color='r', linestyle='--')
# 更新画布显示
plt.draw()

# 等待用户点击窗口或按键后关闭
plt.waitforbuttonpress()
plt.close()

方法二：使用面向对象编程（更推荐） #

显式创建画布和轴对象，所有绘图操作都基于这个固定的轴，彻底避免全局状态的问题：

import matplotlib.pyplot as plt

# 显式创建画布和轴
fig, ax = plt.subplots()
# 在轴上绘制初始曲线
ax.plot([1, 2, 3, 4], [1, 4, 9, 16], 'b-')

# 显示窗口但不阻塞
plt.show(block=False)

# 获取用户输入
y_value = float(input("Enter the y-value for the line: "))

# 在已有轴上添加水平线
ax.axhline(y=y_value, color='r', linestyle='--')
# 更新画布内容
fig.canvas.draw()

# 等待用户操作后关闭窗口
plt.waitforbuttonpress()
plt.close()

关键注意点 #

• 默认非交互式模式下，plt.show()会阻塞到窗口关闭，关闭后画布资源被销毁，无法再更新。
• 不要在plt.show()之后使用plt.xxx这类全局绘图命令，此时全局画布已经不存在了。

内容的提问来源于stack exchange，提问作者S P