如何使用ipywidgets创建支持切换两个numpy数组输入的交互绘图部件？

问题原因 #

你原有代码中array=(a,b)的参数写法不符合ipywidgets的规则：当interact接收到元组参数时，默认会生成连续数值调节滑块，无法识别为可选的数组列表，需要显式声明下拉选择组件来绑定数组选项。

可运行的修正代码 #

import numpy as np
import matplotlib.pyplot as plt
from __future__ import print_function
from ipywidgets import interact, interactive, fixed, interact_manual
import ipywidgets as widgets

# 定义两个数组
a = np.linspace(1,2,100)
b = np.linspace(3,4,100)

def test(array, constant):
    # 向量化计算替代循环，效率更高
    f = array ** 2 + constant * array
    # x轴使用原数组值，和你定义的区间对应，避免默认展示索引值
    plt.plot(array, f)
    plt.xlabel('x值')
    plt.ylabel('计算结果')
    plt.title(f'x²+{constant}x 函数曲线')
    plt.show()
    return f

# 显式创建下拉选择组件，绑定数组选项
array_dropdown = widgets.Dropdown(
    options=[('区间a[1,2]', a), ('区间b[3,4]', b)],
    value=a,
    description='选择区间:'
)

# 绑定交互组件，constant设置0.1步长方便精细调节
interact(test, array=array_dropdown, constant=(1,5,0.1))

改动说明 #

• 新增Dropdown下拉组件，每个选项可以自定义显示名称和对应的实际数组值，切换时会自动把选中的数组传入目标函数
• 优化了绘图逻辑，直接用numpy向量化运算替代循环，同时x轴使用原数组值显示，和你定义的区间对应，避免展示偏差
• 给constant参数增加了0.1的调节步长，调节参数更灵活

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