VS Code中Plotly+ipywidgets切换菜单重复生成图表的问题

问题描述 #

以下代码预期通过下拉菜单切换生成散点图或箱线图：

import plotly.graph_objs as go
import ipywidgets as widgets
import numpy as np

# Create some data for the plots
x = np.random.rand(50)
y = np.random.rand(50)

# Create a function that will be called when the dropdown value changes
def update_plot(plot_type):
    fig = go.Figure()
    
    if plot_type == 'Scatter Plot':
        fig.add_trace(go.Scatter(x=x, y=y, mode='markers'))
    elif plot_type == 'Box Plot':
        fig.add_trace(go.Box(y=y))
    
    fig.show()

# Create a dropdown menu
dropdown = widgets.Dropdown(
    options=['Scatter Plot', 'Box Plot'],
    value='Scatter Plot',
    description='Plot Type:',
)

# Display the dropdown menu
display(dropdown)

# Call the function to update the plot whenever the dropdown value changes
widgets.interactive(update_plot, plot_type=dropdown)

在Jupyter Lab中运行时可正常切换更新已有图表，但在VS Code的.ipynb单元格中，每次切换下拉菜单都会生成新图表而非更新原有图表，Matplotlib实现相同逻辑则无此问题。当前环境：VS Code 1.84.2、plotly 5.18.0、ipywidgets 8.1.1。

修复方案 #

核心改动是使用plotly.graph_objs.FigureWidget替代普通go.Figure，它支持原地更新图表内容，而非每次生成新实例。修改后的完整代码如下：

import plotly.graph_objs as go
import ipywidgets as widgets
import numpy as np

# 生成数据
x = np.random.rand(50)
y = np.random.rand(50)

# 初始化FigureWidget实例，作为可更新的图表容器
fig_widget = go.FigureWidget()

# 更新图表的函数：修改已有FigureWidget的data，而非新建图表
def update_plot(plot_type):
    with fig_widget.batch_update():
        # 清空现有trace
        fig_widget.data = []
        if plot_type == 'Scatter Plot':
            fig_widget.add_trace(go.Scatter(x=x, y=y, mode='markers'))
        elif plot_type == 'Box Plot':
            fig_widget.add_trace(go.Box(y=y))

# 创建下拉菜单
dropdown = widgets.Dropdown(
    options=['Scatter Plot', 'Box Plot'],
    value='Scatter Plot',
    description='Plot Type:',
)

# 绑定交互逻辑
widgets.interactive(update_plot, plot_type=dropdown)

# 显示图表容器和下拉菜单
display(dropdown, fig_widget)

关键改动说明 #

• 使用FigureWidget作为持久化的图表容器，替代每次在函数内新建的go.Figure
• 用batch_update()包裹修改操作，减少渲染次数，提升流畅度
• 直接修改fig_widget.data来更新图表内容，而非调用fig.show()触发新渲染
• 最后统一显示下拉菜单和图表容器，确保交互时仅更新已有图表实例

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