如何将下拉小部件输出作为数据框？

Question

下面是使用两个下拉菜单根据字段和井名过滤数据框的代码。过滤器应用于 pandas 数据帧，我希望过滤后的输出 (common_filter) 也是一种 pandas 数据帧。当前，当是小部件类型时。有没有办法把它作为数据框？

以下代码取自TowardsDataScience，稍作修改。

"unique_sorted_values" 函数只是 returns 传递数组的唯一排序值列表，在本例中为 FieldID 和 WellnameID

import ipywidgets as widgets

# dummy data
df = pd.DataFrame({'FieldID': ['A', 'A', 'A', 'B', 'B', 'B', 'B'],
                        'WellnameID':['1_A', '1_A', '2_A', '1_B', '1_B', '2_B', '2_B'],
                        'value': [1, 2, 3, 4, 5, 6, 7]})

output = widgets.Output()

dropdown_field = widgets.Dropdown(options = unique_sorted_values(df.FieldID))
dropdown_wellname = widgets.Dropdown(options = unique_sorted_values(df[df.FieldID==dropdown_field.value].WellnameID))

def common_filtering(field, wellname):
    output.clear_output()

    common_filter = df[(df.FieldID == field) & (df.WellnameID == wellname)]

    with output:
        display(common_filter)

def dropdown_field_eventhandler(change):
    common_filtering(change.new, dropdown_wellname.value)
def dropdown_wellname_eventhandler(change):
    common_filtering(dropdown_field.value, change.new)

dropdown_field.observe(dropdown_field_eventhandler, names='value')
dropdown_wellname.observe(dropdown_wellname_eventhandler, names='value')

input_widgets = widgets.HBox([dropdown_field, dropdown_wellname])
display(input_widgets)

display(output)

Answer 1

您不能将函数的 return 值用于数据框，因为 return 值未分配给代码主体中的任何内容（它作为打回来）。当您想创建一个全新的数据框（而不是修改现有数据框）时，一种简单的方法是在初始数据的复制版本上使用 global 关键字。

选择下拉菜单后，您应该能够在下面的单元格中获取 filtered 数据框并查看过滤器的影响。如果您需要更复杂的东西，您可能想要构造一个 class 对象来跟踪数据状态、应用过滤器等。

    import ipywidgets as widgets
    import pandas as pd

    # dummy data
    df = pd.DataFrame({'FieldID': ['A', 'A', 'A', 'B', 'B', 'B', 'B'],
                            'WellnameID':['1_A', '1_A', '2_A', '1_B', '1_B', '2_B', '2_B'],
                            'value': [1, 2, 3, 4, 5, 6, 7]})

    filtered = pd.DataFrame()
    output = widgets.Output()

    dropdown_field = widgets.Dropdown(options = sorted(df.FieldID.unique()))
    dropdown_wellname = widgets.Dropdown(options = sorted(df[df.FieldID==dropdown_field.value].WellnameID.unique()))

    def common_filtering(field, wellname):
        global filtered
        output.clear_output()

        filtered = df[(df.FieldID == field) & (df.WellnameID == wellname)]

        with output:
            display(filtered)

    def dropdown_field_eventhandler(change):
        common_filtering(change.new, dropdown_wellname.value)
    def dropdown_wellname_eventhandler(change):
        common_filtering(dropdown_field.value, change.new)

    dropdown_field.observe(dropdown_field_eventhandler, names='value')
    dropdown_wellname.observe(dropdown_wellname_eventhandler, names='value')

    input_widgets = widgets.HBox([dropdown_field, dropdown_wellname])
    display(input_widgets)

    display(output)

如何将下拉小部件输出作为数据框？

How to get dropdown widget output as a dataframe?

python

ipywidgets