Question

这是使用两个下拉列表根据字段和井名过滤数据框的代码。该过滤器应用于pandas数据框，我希望过滤后的输出（common_filter）也成为pandas数据框的一种。当前，当is为窗口小部件类型时。有什么办法可以将其作为数据框吗？

下面的代码摘自TowardsDataScience，并进行了一些修改。

“ unique_sorted_values”函数仅返回已传递数组的唯一排序值列表，在这种情况下为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值用于数据框，因为返回值未分配给代码主体中的任何内容（它作为回调传递给交互）。当您要创建一个全新的数据框（而不是修改现有的数据框）时，一种简单的方法是在复制的初始数据版本上使用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)

如何获取下拉窗口小部件输出作为数据框？

1 个答案: