如何在VS Code Notebook中为Pandas Dataframe创建交互式列选择面板

直接用ipywidgets库就能实现可视化选列，无需手动输入列名，以下是两种实用的实现方式：

1. 多选下拉框（适合列数较多的场景） #

这种方式通过下拉多选组件快速选择列，支持批量操作：

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

# 模拟带长列名、含空格的测试DataFrame
data = {
    "用户姓名 完整信息": ["张三", "李四", "王五"],
    "订单编号 唯一标识": ["OD20240501001", "OD20240501002", "OD20240501003"],
    "购买日期 精确到时分": ["2024-05-01 10:30", "2024-05-01 14:15", "2024-05-02 09:45"],
    "商品类别 详细分类": ["电子产品-手机", "家居用品-厨具", "服饰-外套"],
    "支付金额 人民币": [3999, 299, 599],
    "物流状态 实时更新": ["已发货", "配送中", "已签收"]
}
df = pd.DataFrame(data)

# 创建多选下拉组件，适配长列名显示
column_selector = widgets.SelectMultiple(
    options=df.columns.tolist(),
    value=[df.columns[0]],  # 默认选中第一列
    description='选择列:',
    layout=widgets.Layout(width='500px', height='200px')
)

# 定义列选择后的展示函数
def show_selected_columns(selected_cols):
    if selected_cols:
        display(df[list(selected_cols)])
    else:
        print("请至少选择一列")

# 绑定组件与展示逻辑
widgets.interact(show_selected_columns, selected_cols=column_selector);

运行后会出现一个多选框，按住Ctrl（Windows）或Cmd（Mac）可多选列，选中后下方会自动展示仅包含所选列的DataFrame。

2. 复选框组（适合列数较少的场景） #

如果DataFrame列数不多，用复选框组更直观：

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

# 复用上面的测试DataFrame（如果已创建可跳过）
# df = pd.DataFrame(data)

# 为每一列创建复选框
checkbox_list = [widgets.Checkbox(value=False, description=col) for col in df.columns]
checkbox_box = widgets.VBox(checkbox_list)

# 定义复选框变化后的处理逻辑
def update_display(change):
    selected_cols = [cb.description for cb in checkbox_list if cb.value]
    if selected_cols:
        display(df[selected_cols])
    else:
        print("请至少选择一列")

# 为每个复选框绑定事件
for cb in checkbox_list:
    cb.observe(update_display, names='value')

# 显示复选框组
display(checkbox_box)

前置要求 #

• 先安装依赖库：pip install ipywidgets pandas
• VS Code需安装Jupyter扩展，且Notebook使用的Python内核已安装上述库

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