Python函数中Pandas DataFrame可视化编辑与保存方案咨询

1. 使用tkinter结合ttk.Treeview（原生库，无需额外安装） #

直接用Python自带的tkinter构建轻量编辑界面，修改操作实时同步回DataFrame，函数结束后可直接获取修改结果。示例代码：

import tkinter as tk
from tkinter import ttk
import pandas as pd

def edit_dataframe(df):
    root = tk.Tk()
    root.title("DataFrame 编辑器")
    
    # 创建表格组件
    tree = ttk.Treeview(root)
    tree["columns"] = list(df.columns)
    tree["show"] = "headings"
    
    # 设置表头
    for col in df.columns:
        tree.heading(col, text=col)
        tree.column(col, width=100)
    
    # 填充初始数据
    for index, row in df.iterrows():
        tree.insert("", "end", values=list(row))
    
    # 双击单元格触发编辑
    def edit_cell(event):
        item = tree.selection()[0]
        col = tree.identify_column(event.x)
        col_idx = int(col.replace("#", "")) - 1
        old_value = tree.item(item, "values")[col_idx]
        
        # 创建临时输入框
        entry = ttk.Entry(root)
        entry.place(x=event.x, y=event.y)
        entry.insert(0, old_value)
        
        # 回车保存修改
        def save_edit(event):
            new_value = entry.get()
            tree.set(item, column=df.columns[col_idx], value=new_value)
            df.iloc[int(item.split("I")[1]), col_idx] = new_value
            entry.destroy()
        
        entry.bind("<Return>", save_edit)
        entry.focus()
    
    tree.bind("<Double-1>", edit_cell)
    
    # 关闭窗口时结束进程
    def on_close():
        root.destroy()
    
    root.protocol("WM_DELETE_WINDOW", on_close)
    tree.pack(fill="both", expand=True)
    root.mainloop()
    return df

# 测试用例
if __name__ == "__main__":
    sample_df = pd.DataFrame({
        "Name": ["Alice", "Bob"],
        "Age": [25, 30],
        "City": ["New York", "London"]
    })
    modified_df = edit_dataframe(sample_df)
    print("修改后的DataFrame:\n", modified_df)

2. 使用ipywidgets（适合Jupyter环境） #

在Jupyter Notebook/Lab场景下，用ipywidgets的DataGrid组件支持直接编辑，点击保存按钮后即可同步更新DataFrame。示例代码：

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

def edit_dataframe_jupyter(df):
    grid = widgets.DataGrid(df, editable=True)
    display(grid)
    
    # 保存修改的触发函数
    def update_df(b):
        nonlocal df
        df = pd.DataFrame(grid.value)
        print("修改已同步到DataFrame")
    
    save_btn = widgets.Button(description="保存修改")
    save_btn.on_click(update_df)
    display(save_btn)
    return df

# 测试用例
sample_df = pd.DataFrame({
    "Name": ["Alice", "Bob"],
    "Age": [25, 30],
    "City": ["New York", "London"]
})
modified_df = edit_dataframe_jupyter(sample_df)

3. 使用PyQt5/PyQt6的QTableWidget #

需要更专业的桌面级GUI时，可选择PyQt构建编辑器，支持更丰富的交互逻辑，修改后的数据会自动匹配原DataFrame的类型。示例代码（PyQt5）：

import sys
import pandas as pd
from PyQt5.QtWidgets import (QApplication, QMainWindow, QTableWidget, 
                             QTableWidgetItem, QVBoxLayout, QWidget)

class DataFrameEditor(QMainWindow):
    def __init__(self, df, parent=None):
        super().__init__(parent)
        self.df = df.copy()
        self.init_ui()
    
    def init_ui(self):
        self.setWindowTitle("DataFrame 编辑器")
        self.table_widget = QTableWidget()
        self.load_table_data()
        
        layout = QVBoxLayout()
        layout.addWidget(self.table_widget)
        
        central_widget = QWidget()
        central_widget.setLayout(layout)
        self.setCentralWidget(central_widget)
    
    def load_table_data(self):
        rows, cols = self.df.shape
        self.table_widget.setRowCount(rows)
        self.table_widget.setColumnCount(cols)
        self.table_widget.setHorizontalHeaderLabels(self.df.columns)
        
        for i in range(rows):
            for j in range(cols):
                self.table_widget.setItem(i, j, QTableWidgetItem(str(self.df.iloc[i, j])))
    
    def get_updated_df(self):
        # 读取表格数据并转换回原数据类型
        rows, cols = self.df.shape
        for i in range(rows):
            for j in range(cols):
                item = self.table_widget.item(i, j)
                if item:
                    original_type = type(self.df.iloc[i, j])
                    try:
                        self.df.iloc[i, j] = original_type(item.text())
                    except ValueError:
                        # 转换失败则保留原数据
                        pass
        return self.df

def edit_dataframe(df):
    app = QApplication(sys.argv)
    editor = DataFrameEditor(df)
    editor.show()
    app.exec_()
    return editor.get_updated_df()

# 测试用例
if __name__ == "__main__":
    sample_df = pd.DataFrame({
        "Name": ["Alice", "Bob"],
        "Age": [25, 30],
        "City": ["New York", "London"]
    })
    modified_df = edit_dataframe(sample_df)
    print("修改后的DataFrame:\n", modified_df)

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