python - Jupyter 笔记本小部件 : Create dependent dropdowns

Question

我想在我的 Jupyter Notebook 中创建 2 个下拉小部件。下拉内容取自数据框。

假设我有一个由 3 个分类变量“a”、“b”、“c”组成的 pandas 数据框。 'a' 有 3 个子类型 'a1'、'a2' 和 'a3'。 'b' 和 'c' 与 a 相似，因为它们也有自己的子类型。我想创建 2 个下拉小部件:第一个下拉小部件将具有 ['a','b','c']，第二个下拉小部件将根据用户为第一个小部件选择的变量显示子类型。

老实说，我不知道该怎么做。我将尝试为此编写一些代码:

import pandas as pd
from IPython.display import *
import ipywidgets as widgets
from ipywidgets import *

# Create the dataframe
df = pd.DataFrame([['a1','a2','a3'],
             ['b1','b2','b3'],
             ['c1','c2','c3']], index = ['a','b','c']).transpose()

# Widgets
widget1 = Dropdown(options = ['a','b','c'])
display(widget1)
widget2 = Dropdown(???????)
display(widget2)

根据我为两个下拉小部件选择的内容，我希望执行一些功能。

感谢任何帮助。

Answer 1

我找到了如何做到这一点。我希望这对其他也想做同样事情的人有所帮助。

x_widget = Dropdown(options = ['a','b','c'])
y_widget = Dropdown()

# Define a function that updates the content of y based on what we select for x
def update(*args):
    y_widget.options = df[x_widget.value].unique().tolist()
x_widget.observe(update)

# Some function you want executed
def random_function():
...

interact(random_function,
         x = x_widget,
         y = y_widget);