使用IPython进行双变量交互式函数绘图

时间:2015-03-02 12:49:26

标签: python ipython ipython-notebook

我想用不同的频率绘制两个函数,比如正弦和余弦 - 所以第一个变量是绘制函数,第二个变量是频率。我想要一个选择器功能的选择器小部件和一个选择频率的滑块。是否可以使用interact实现此目的,还是需要更复杂的设置?

1 个答案:

答案 0 :(得分:2)

是的,interact这应该是可能的。为了进一步阅读,github存储库中有几个example notebooks可以用作交互式小部件的介绍。

%matplotlib inline
from IPython.html import widgets
import numpy as np
import matplotlib.pyplot as plt

fun_map = {
    "sin": np.sin,
    "cos": np.cos
}

func_name = widgets.Dropdown(
    options=['sin', 'cos'],
    value='sin',
    description='Function:',
)

freq = widgets.FloatSlider(
    min=1,
    max=5,
    value=1,
    description='Parameter:'
)

def plot_fun(func_name, freq, fun_map):
    f = fun_map[func_name]
    x = np.linspace(0, 2*np.pi, 100)
    plt.plot(x, f(freq * x))

res = widgets.interact(plot_fun, freq=freq, func_name=func_name,
                       fun_map=widgets.fixed(fun_map))

结果如下:

result