点击切换图表后保持Matplotlib按钮可见的实现方案

核心问题是plt.clf()会清除整个画布的所有元素（包括按钮），所以要避免直接清除画布，改为只清除图表所在的轴内容，同时将按钮放在独立的轴上，确保切换操作不影响按钮。

方法1：分离图表轴与按钮轴 #

通过创建两个独立的轴，一个用于展示图表，一个放置按钮，切换时仅更新图表轴的内容：

import matplotlib.pyplot as plt
from matplotlib.widgets import Button

# 创建画布和图表轴
fig, ax_chart = plt.subplots()
# 调整布局，为按钮预留底部空间
plt.subplots_adjust(bottom=0.2)

# 创建按钮轴（位置参数：[左偏移, 下偏移, 宽度, 高度]）
ax_btn = plt.axes([0.4, 0.05, 0.2, 0.075])
toggle_btn = Button(ax_btn, '切换图表')

# 示例数据
x = [1, 2, 3, 4]
y_line1 = [1, 4, 9, 16]
y_line2 = [1, 2, 3, 4]
current_chart = 1

def switch_chart(event):
    global current_chart
    # 仅清除图表轴的内容，不影响按钮轴
    ax_chart.clear()
    if current_chart == 1:
        ax_chart.plot(x, y_line2)
        current_chart = 2
    else:
        ax_chart.plot(x, y_line1)
        current_chart = 1
    # 重新绘制图表
    plt.draw()

# 绑定按钮点击事件
toggle_btn.on_clicked(switch_chart)

# 初始绘制第一个图表
ax_chart.plot(x, y_line1)
plt.show()

方法2：复用当前图表轴清除内容 #

如果不想手动创建多个轴，可以通过plt.gca()获取当前图表轴，仅清除轴内元素而非整个画布：

import matplotlib.pyplot as plt
from matplotlib.widgets import Button

fig = plt.figure()
plt.subplots_adjust(bottom=0.2)

# 创建按钮轴
ax_btn = plt.axes([0.4, 0.05, 0.2, 0.075])
toggle_btn = Button(ax_btn, '切换图表')

x = [1, 2, 3, 4]
y_line1 = [1, 4, 9, 16]
y_line2 = [1, 2, 3, 4]
current_chart = 1

def switch_chart(event):
    global current_chart
    # 获取当前活跃的图表轴
    ax = plt.gca()
    # 清除轴内所有绘图元素
    ax.clear()
    if current_chart == 1:
        ax.plot(x, y_line2)
        current_chart = 2
    else:
        ax.plot(x, y_line1)
        current_chart = 1
    plt.draw()

toggle_btn.on_clicked(switch_chart)

plt.plot(x, y_line1)
plt.show()

原理说明 #

• plt.clf()会清除整个画布的所有轴和元素，包括按钮所在的轴，导致按钮消失。
• 改用ax.clear()仅清除指定轴内的绘图内容（线条、标签等），按钮所在的独立轴不受影响，因此能始终保持显示。

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