修复ipywidgets函数：用户选择未写入字典，仅存默认值

修正后的代码 #

import pandas as pd
from ipywidgets import interact, widgets, Button, VBox
from functools import partial

def make_NA_dict(df, options):
    # 获取所有需要补全的去重Matter Number
    list_of_NAs = df[df['Practice Area'].isna()]['Matter Number'].unique().tolist()
    NA_dict = {}

    def capture_choice(matter, option):
        NA_dict[matter] = option

    widgets_list = []
    for matter in list_of_NAs:
        label = widgets.Label(f"Choose practice for {matter}:")
        dropdown = widgets.Dropdown(options=options, value="Other", description="")
        # 用lambda默认参数绑定当前循环的matter，避免变量引用错误
        dropdown.observe(lambda change, m=matter: capture_choice(m, change['new']), names='value')
        widgets_list.append(VBox([label, dropdown]))

    # 创建完成按钮替代input()，避免阻塞事件循环
    done_btn = Button(description="Done")
    widgets_list.append(done_btn)

    # 渲染所有控件
    display(VBox(widgets_list))

    # 等待按钮点击
    btn_clicked = False
    def on_done_click(_):
        nonlocal btn_clicked
        btn_clicked = True

    done_btn.on_click(on_done_click)

    # 非阻塞等待完成信号
    import time
    while not btn_clicked:
        time.sleep(0.1)

    return NA_dict

问题根源与修改说明 #

1. 循环变量引用错误
   原代码中循环内的lambda共享同一个matter变量引用，当用户选择选项时，循环已结束，matter会固定为最后一个值，导致所有选择都被错误记录到最后一条Matter Number上，且初始默认值会直接写入字典。修改时通过lambda默认参数m=matter绑定当前循环的变量值，确保每个回调对应正确的Matter Number。

2. interact自动触发初始调用
   interact在创建时会自动调用一次回调，直接把默认值"Other"写入字典。改用observe监听下拉框值变化事件，仅在用户主动修改时触发回调，避免初始值误写入。

3. input()阻塞事件循环
   Jupyter环境中input()会阻塞UI事件循环，导致widget交互无法实时响应。改用ipywidgets的按钮组件触发完成信号，配合非阻塞等待逻辑，保证交互正常运行。

示例使用 #

# 测试用DataFrame
df = pd.DataFrame({
    'Matter Number': ['MAT-001', 'MAT-002', 'MAT-001', 'MAT-003'],
    'Practice Area': [None, None, 'Litigation', None]
})
options = ['Litigation', 'Corporate', 'IP', 'Other']

# 调用函数并获取结果
result_dict = make_NA_dict(df, options)
print(result_dict)

内容的提问来源于stack exchange，提问作者VSm