从python中的函数更新ipywidget下拉列表

时间:2017-12-26 14:09:21

标签: python widget dropdown ipywidgets

我是Python的新手,我想从ipywidget创建一个交互式下拉列表。主要目的是基于另外两个小部件更新下拉列表。在下面的代码中,小工具 plotType 将根据小工具 headers_x headers_y 的输入进行更新(两者都指向所选的数据框列)用于绘图)。如果 headers_x headers_y 都包含选择选项,那么 plotType 需要显示" 选择"。但是,如果 headers_x headers_y 选择了其他选项(数据框中的列),则 plotType 需要相应更改。如果 headers_x headers_y 都是数字,那么 plotType 需要显示: numericVsNumeric ,但是如果 headers_x 是分类, headers_y 是数字,然后 plotType 需要显示' catergoricalVsNumeric '我尝试了如下解决方案,但plotType小部件中的选项不会更新。任何帮助深表感谢。谢谢。

from ipywidgets import *
import seaborn.apionly as sns
df = sns.load_dataset('iris')

#identifies the columns in the dataframe
df_cols = list(df.columns.values)
df_cols.insert(0, 'Select')
str_cols = list(df.select_dtypes(include=['object']).columns.values)
str_cols.insert(0, 'Select')

#plot function
def set_plot(headers_x, headers_y, plotType):
    data = df
    #plotting functions to be added

#function to specify the type of plot based on users input
def set_plotType():
    data = df
        #If no selection has been made
    if headers_x.value == 'Select' and headers_y.value == 'Select':
        init = list(['Make Selection'])
    else:
        #if x and y are both numeric
        if data[headers_x.value].dtype == np.float and data[headers_y.value].dtype == np.float:
            init = list(['NumericVsNumeric'])
            #if x is categorical and y is numeric
        elif data[headers_x.value].dtype == np.object and data[headers_y.value].dtype == np.float:
            init = list(['CategoricalVsNumeric'])

    return init


#define widgets
headers_x = widgets.Dropdown(
        options=df_cols,
        value=df_cols[0],
        description='X'
    )

headers_x.set_title  = 'headers_x'

headers_y = widgets.Dropdown(
        options=df_cols,
        value=df_cols[0],
        description='Y'
    )

headers_y.set_title  = 'headers_y'

plotType = widgets.Dropdown(
        options=set_plotType(),
        #value=df_cols[0],
        description='Plot Type'
    )

plotType.set_title  = 'plotType'


#interact function
interact(set_plot, headers_x = headers_x, headers_y = headers_y, plotType = plotType)

1 个答案:

答案 0 :(得分:2)

我通过使用观察实现了这一点。这意味着只要您的前两个下拉选项发生更改,它们就会运行set_Plottype函数。

我将headers.x和headers.y更改为OR,因为您需要同时定义。

当x为数字且y为分类时,我还给了你第三个选项。

from ipywidgets import *
import numpy as np
import seaborn.apionly as sns
df = sns.load_dataset('iris')

#identifies the columns in the dataframe
df_cols = list(df.columns.values)
df_cols.insert(0, 'Select')
str_cols = list(df.select_dtypes(include=['object']).columns.values)
str_cols.insert(0, 'Select')

#plot function
def set_plot(headers_x, headers_y, plotType):
    data = df
    #plotting functions to be added

#function to specify the type of plot based on users input
def set_plotType(_):
    data = df
        #If no selection has been made
    if headers_x.value == 'Select' or headers_y.value == 'Select':
        plotType.options = list(['Make Selection'])
    else:
        #if x and y are both numeric
        if data[headers_x.value].dtype == np.float and data[headers_y.value].dtype == np.float:
            plotType.options = list(['NumericVsNumeric'])
            #if x is categorical and y is numeric
        elif data[headers_x.value].dtype == np.object and data[headers_y.value].dtype == np.float:
            plotType.options = list(['CategoricalVsNumeric'])
        elif data[headers_x.value].dtype == np.float and data[headers_y.value].dtype == np.object:
            plotType.options = list(['NumericalVsCategoric'])



#define widgets
headers_x = widgets.Dropdown(
        options=df_cols,
        value=df_cols[0],
        description='X'
    )

headers_x.set_title  = 'headers_x'

headers_y = widgets.Dropdown(
        options=df_cols,
        value=df_cols[0],
        description='Y'
    )

headers_y.set_title  = 'headers_y'

plotType = widgets.Dropdown(
        options=[],
        description='Plot Type'
    )

headers_x.observe(set_plotType)
headers_y.observe(set_plotType)


#interact function
interact(set_plot, headers_x = headers_x, headers_y = headers_y, plotType = plotType)

enter image description here