PyQt中如何同时为QTableView设置过滤器与添加QPushButton？

你遇到的核心矛盾是代理模型（QSortFilterProxyModel）的索引与原始模型索引不匹配：

• 直接给原始模型索引绑定按钮（setIndexWidget）时，切换到代理模型后，视图使用的是代理索引，找不到之前绑定的按钮；
• 用原始模型时，代理的过滤逻辑无法生效。

下面提供两种可行的解决方案，分别适配不同场景：

方案1：自定义委托实现按钮（推荐，适配所有模型） #

通过QStyledItemDelegate在最后一列绘制按钮，这种方式与视图绑定，不管模型是原始还是代理，都能正常显示按钮并响应点击，性能也更优，适合大数据量场景。

完整代码示例 #

import sys
from PyQt5.QtWidgets import (QApplication, QMainWindow, QLineEdit, QTableView, 
                             QStyledItemDelegate, QVBoxLayout, QWidget)
from PyQt5.QtGui import QPainter, QPalette
from PyQt5.QtCore import Qt, QSortFilterProxyModel, QAbstractTableModel, QModelIndex, QSize

# 自定义表格模型
class MyTableModel(QAbstractTableModel):
    def __init__(self, data, parent=None):
        super().__init__(parent)
        self._data = data
        self._headers = ["姓名", "年龄", "操作"]

    def rowCount(self, parent=QModelIndex()):
        return len(self._data)

    def columnCount(self, parent=QModelIndex()):
        return len(self._headers)

    def data(self, index, role=Qt.DisplayRole):
        if not index.isValid():
            return None
        row = index.row()
        col = index.column()
        if role == Qt.DisplayRole and col < 2:
            return self._data[row][col]
        return None

    def headerData(self, section, orientation, role=Qt.DisplayRole):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self._headers[section]
        return None

# 自定义按钮委托
class ButtonDelegate(QStyledItemDelegate):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.button_clicked = None  # 点击回调函数

    def paint(self, painter, option, index):
        # 仅在最后一列绘制按钮
        if index.column() != 2:
            super().paint(painter, option, index)
            return
        
        # 绘制按钮样式（模拟原生按钮状态）
        painter.save()
        if option.state & QStyle.State_MouseOver:
            painter.fillRect(option.rect, option.palette.highlight())
        elif option.state & QStyle.State_Selected:
            painter.fillRect(option.rect, option.palette.dark())
        else:
            painter.fillRect(option.rect, option.palette.button())
        
        # 绘制按钮文字
        painter.setPen(option.palette.buttonText().color())
        painter.drawText(option.rect, Qt.AlignCenter, "操作")
        painter.restore()

    def sizeHint(self, option, index):
        # 设置按钮固定大小
        if index.column() == 2:
            return QSize(80, 30)
        return super().sizeHint(option, index)

    def editorEvent(self, event, model, option, index):
        # 处理鼠标点击事件
        if index.column() == 2 and event.type() == event.MouseButtonRelease and event.button() == Qt.LeftButton:
            if self.button_clicked:
                # 将代理索引映射回原始模型索引（兼容代理模型场景）
                source_index = index
                if isinstance(model, QSortFilterProxyModel):
                    source_index = model.mapToSource(index)
                self.button_clicked(source_index.row())
            return True
        return super().editorEvent(event, model, option, index)

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("过滤+按钮示例")
        self.resize(500, 300)

        # 初始化测试数据
        self.raw_data = [
            ["张三", 25],
            ["李四", 30],
            ["王五", 22],
            ["赵六", 35],
            ["张三疯", 28]
        ]

        # 创建组件
        self.search_edit = QLineEdit()
        self.search_edit.setPlaceholderText("输入姓名过滤")
        self.table = QTableView()

        # 布局
        central_widget = QWidget()
        layout = QVBoxLayout(central_widget)
        layout.addWidget(self.search_edit)
        layout.addWidget(self.table)
        self.setCentralWidget(central_widget)

        # 设置模型与代理
        self.source_model = MyTableModel(self.raw_data)
        self.proxy_model = QSortFilterProxyModel()
        self.proxy_model.setSourceModel(self.source_model)
        self.proxy_model.setFilterKeyColumn(0)  # 按第一列（姓名）过滤

        # 绑定自定义委托到最后一列
        self.button_delegate = ButtonDelegate()
        self.button_delegate.button_clicked = self.on_button_clicked
        self.table.setItemDelegateForColumn(2, self.button_delegate)

        # 表格使用代理模型，同时启用过滤和按钮
        self.table.setModel(self.proxy_model)

        # 搜索框输入绑定过滤逻辑
        self.search_edit.textChanged.connect(self.proxy_model.setFilterFixedString)

    def on_button_clicked(self, row):
        # 按钮点击后的业务逻辑
        print(f"点击了第{row+1}行按钮，数据：{self.raw_data[row]}")

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

关键说明 #

• 委托通过paint方法绘制按钮样式，editorEvent处理点击事件，与模型类型无关；
• 点击时通过mapToSource将代理索引转换为原始模型索引，确保能获取正确的原始数据。

方案2：setIndexWidget配合代理模型信号更新按钮 #

如果习惯直接使用setIndexWidget，需要监听代理模型的内容变化信号，每次过滤结果更新时，重新给代理模型的索引绑定按钮，适合小数据量场景。

核心代码片段 #

class MainWindow(QMainWindow):
    def __init__(self):
        # ... 其他初始化代码同方案1，移除委托相关部分 ...

        # 表格使用代理模型
        self.table.setModel(self.proxy_model)

        # 绑定搜索框与代理模型的更新信号，触发按钮刷新
        self.search_edit.textChanged.connect(self.update_buttons)
        self.proxy_model.rowsInserted.connect(self.update_buttons)
        self.proxy_model.rowsRemoved.connect(self.update_buttons)
        self.proxy_model.layoutChanged.connect(self.update_buttons)

        # 初始加载按钮
        self.update_buttons()

    def update_buttons(self):
        # 先清空已有按钮，避免重复创建
        for row in range(self.table.rowCount()):
            index = self.table.model().index(row, 2)
            self.table.setIndexWidget(index, None)
        
        # 给代理模型的每一行最后一列添加按钮
        for row in range(self.table.rowCount()):
            proxy_index = self.proxy_model.index(row, 2)
            btn = QPushButton("操作")
            # 映射回原始模型索引，确保点击时获取正确数据
            source_index = self.proxy_model.mapToSource(proxy_index)
            btn.clicked.connect(lambda checked, r=source_index.row(): self.on_button_clicked(r))
            self.table.setIndexWidget(proxy_index, btn)

    def on_button_clicked(self, row):
        print(f"点击了第{row+1}行按钮，数据：{self.raw_data[row]}")

关键说明 #

• 监听代理模型的rowsInserted、rowsRemoved、layoutChanged信号，每次过滤结果变化时重新绑定按钮；
• 按钮点击时通过mapToSource转换索引，确保获取原始数据的行号。

内容的提问来源于stack exchange，提问作者J Chandra Kiran