为何Matplotlib交互式绘图代码放入函数后失去交互功能？

1. 对象被垃圾回收：函数执行完毕后，fig、button等交互控件对象没有外部引用，被Python回收，导致按钮点击无响应。
2. 全局变量误用：button_clicked函数里声明global line_plotted，但全局作用域并未定义该变量，而函数内的line_plotted是局部变量，导致状态逻辑异常。
3. 交互模式使用错误：开启plt.ion()后，plt.show()会立即返回，函数结束后事件循环终止，交互功能无法维持。

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

def make_plot():
    # 用列表存储状态，无需global/nonlocal即可修改
    line_plotted = [True]
    
    def button_clicked(event):
        if line_plotted[0]:
            ax1.lines[1].remove()
            line_plotted[0] = False
        else:
            ax1.plot(line[0], line[1], color='r', linestyle='--')
            line_plotted[0] = True
        fig.canvas.draw_idle()
    
    x = np.linspace(0, 4*np.pi, 500)
    y = np.sin(x)
    line = ([0, 4 * np.pi], [1, 1])
    
    fig = plt.figure(figsize=(15,5))
    ax1 = fig.add_subplot()
    ax1.plot(x,y)
    ax1.plot(line[0], line[1], color='r', linestyle='--')
    
    button_ax = plt.axes([0.8, 0.05, 0.1, 0.075])
    button = Button(button_ax, 'Toggle red line')
    button.on_clicked(button_clicked)
    
    # 阻塞窗口直到关闭，维持交互事件循环
    plt.show(block=True)
    
    # 返回对象避免被回收
    return fig, button

fig, btn = make_plot()

• 状态存储优化：用列表line_plotted存储状态，利用列表的可变性，让内部函数直接修改状态值，避开全局/局部变量的作用域问题。
• 维持事件循环：plt.show(block=True)会保持窗口打开并运行事件循环，直到用户手动关闭窗口，确保交互事件能被正常处理。
• 高效绘图更新：用fig.canvas.draw_idle()替代draw()，只更新需要重绘的区域，提升性能。
• 防止对象回收：返回fig和button给外部变量引用，确保这些控件对象不会被Python垃圾回收机制清理。

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