如何为pandastable编辑单元格配置可选值下拉菜单

通过继承pandastable.Table类重写单元格编辑触发逻辑，替换默认输入框为下拉选择框即可实现需求，修改后的完整可运行代码如下：

import tkinter as tk
from tkinter import ttk
import sqlite3
from pandastable import Table
import pandas as pd

# 自定义带下拉编辑功能的Table子类
class DropdownTable(Table):
    def __init__(self, parent=None, dataframe=None, column_options=None, **kwargs):
        super().__init__(parent, dataframe=dataframe, **kwargs)
        # 接收各列合法可选值配置，格式为 {列名: [可选值列表]}
        self.column_options = column_options or {}

    def drawCellEntry(self, row, col, text=None):
        # 获取当前编辑的列名
        col_name = self.model.df.columns[col]
        # 仅对配置了可选值的列渲染下拉框
        if col_name in self.column_options:
            options = self.column_options[col_name]
            current_val = self.model.getValueAt(row, col)
            # 创建只读下拉框，只能选择预设值
            self.entry = ttk.Combobox(self.parentframe, values=options, state="readonly")
            # 默认值为单元格当前值
            self.entry.set(current_val)
            
            # 绑定选中回调，选中后自动回填数据
            def on_select(event):
                selected_val = self.entry.get()
                # 保持值的类型和原DataFrame列的类型一致，避免数字转字符串的问题
                col_dtype = self.model.df[col_name].dtype
                selected_val = col_dtype.type(selected_val)
                # 更新底层DataFrame数据
                self.model.setValueAt(selected_val, row, col)
                # 刷新单元格显示
                self.drawCell(row, col)
                # 销毁下拉框组件
                self.entry.destroy()
            
            self.entry.bind("<<ComboboxSelected>>", on_select)
            # 将下拉框定位到当前单元格位置
            x1, y1, x2, y2 = self.getCellCoords(row, col)
            self.entry.place(x=x1, y=y1, width=x2-x1, height=y2-y1)
            self.entry.focus_set()
        # 未配置可选值的列走原生普通编辑逻辑
        else:
            super().drawCellEntry(row, col, text)

def save():
    df.to_sql("crawled", conn, if_exists="replace")
    print(df)
    root.quit()

root = tk.Tk()
root.geometry("1500x1000")

frame = tk.Frame(root)
frame.pack(fill="both", expand=True)

df = pd.DataFrame()
df["column1"] = [1,2,3]
df["column2"] = ["a","b","c"]

conn = sqlite3.connect("99_data_increment.db")

# 配置各列的合法可选值
column_options = {
    "column1": [1, 2, 3],
    "column2": ["a", "b", "c"]
}
# 实例化自定义的DropdownTable
pt = DropdownTable(frame, dataframe=df, width=300, column_options=column_options)
pt.show()

button = tk.Button(root, text="Save edits", command=save)
button.pack()

root.mainloop()

核心逻辑说明 #

• 新增column_options参数灵活配置各列的可选值列表，无需修改核心逻辑即可扩展到更多列
• 重写的drawCellEntry方法会在双击单元格进入编辑状态时自动触发，对配置了可选值的列自动渲染下拉选择框
• 下拉框设置为只读模式，用户只能选择预设的合法值，避免非法输入
• 自动匹配原DataFrame列的类型，不会出现数字被转为字符串的类型不一致问题
• 未配置可选值的列保留原有编辑逻辑，不影响原有功能使用

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