如何在自定义TableModel的QTableView特定单元格插入按钮部件？

问题背景 #

现有QTableView具备以下特性：

• 自定义列表头
• 仅允许单行选中
• 选中行不可编辑
  需要为每行的最后一列添加按钮，此前参考的方案基于QStandardItemModel，不适用于当前继承自QAbstractTableModel的场景。

实现步骤 #

1. 修正模型列数匹配 #

原代码中模型列数与表头列数不匹配（数据3列、表头4列），需让模型列数与表头一致，确保最后一列有对应索引位置：

def columnCount(self, index):
    return len(self.columns)  # 返回表头列数，而非数据子列表长度

2. 实现选中行不可编辑 #

在模型中重写flags方法，根据行的选中状态控制编辑权限：

def flags(self, index):
    base_flags = Qt.ItemIsSelectable | Qt.ItemIsEnabled
    # 获取关联视图，判断当前行是否被选中
    view = self.parent()
    if view and view.selectionModel().isRowSelected(index.row(), QtCore.QModelIndex()):
        return base_flags  # 选中行不可编辑
    return base_flags | Qt.ItemIsEditable  # 未选中行可编辑（无需编辑可直接返回base_flags）

注意：设置模型时需传入视图作为父对象：self.model = TableModel(data, parent=self.table)，同时修改模型__init__方法接收parent参数。

3. 为每行最后一列添加按钮 #

在MainWindow初始化阶段，遍历所有行，用QTableView的setIndexWidget方法添加按钮并绑定点击事件：

def addRowButtons(self):
    last_col = self.model.columnCount(QtCore.QModelIndex()) - 1
    row_count = self.model.rowCount(QtCore.QModelIndex())
    for row in range(row_count):
        btn = QtWidgets.QPushButton("操作")
        # 绑定点击事件，显式传递当前行号避免变量捕获问题
        btn.clicked.connect(lambda checked, r=row: self.onButtonClick(r))
        idx = self.model.index(row, last_col)
        self.table.setIndexWidget(idx, btn)

同时添加按钮点击的槽函数：

def onButtonClick(self, row):
    print(f"点击了第{row+1}行的操作按钮")

4. 完善模型数据显示逻辑 #

最后一列无需显示数据，在data方法中处理：

def data(self, index, role):
    if role == Qt.DisplayRole:
        # 最后一列返回空，留给按钮
        if index.column() == len(self.columns)-1:
            return ""
        # 确保列索引不超出数据范围
        if index.column() < len(self._data[0]):
            return self._data[index.row()][index.column()]
        return ""

完整修改后的代码 #

import sys
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import Qt
from PyQt5.QtWidgets import QAbstractItemView, QHeaderView

class TableModel(QtCore.QAbstractTableModel):
    def __init__(self, data, parent=None):
        self.columns = ["A","B","C","D"]
        super(TableModel, self).__init__(parent)
        self._data = data

    def data(self, index, role):
        if role == Qt.DisplayRole:
            # 最后一列不显示数据，留给按钮
            if index.column() == len(self.columns)-1:
                return ""
            # 确保列索引不超出数据范围
            if index.column() < len(self._data[0]):
                return self._data[index.row()][index.column()]
            return ""

    def rowCount(self, index):
        return len(self._data)

    def columnCount(self, index):
        # 返回表头列数，保证最后一列存在
        return len(self.columns)

    def headerData(self, section: int, orientation: QtCore.Qt.Orientation, role: int = ...):
        if orientation == Qt.Horizontal and role == Qt.DisplayRole:
            return self.columns[section]
        if orientation == Qt.Vertical and role == Qt.DisplayRole:
            return f"{section + 1}"

    def flags(self, index):
        base_flags = Qt.ItemIsSelectable | Qt.ItemIsEnabled
        # 获取关联的视图，判断当前行是否被选中
        view = self.parent()
        if view and view.selectionModel().isRowSelected(index.row(), QtCore.QModelIndex()):
            # 选中行不可编辑
            return base_flags
        # 未选中行可编辑（如果不需要编辑可直接返回base_flags）
        return base_flags | Qt.ItemIsEditable


class MainWindow(QtWidgets.QMainWindow):
    def __init__(self):
        super().__init__()

        self.table = QtWidgets.QTableView()
        self.table.verticalHeader().show()
        self.table.horizontalHeader().show()
        self.table.setSelectionBehavior(QAbstractItemView.SelectRows)
        self.table.setSelectionMode(QAbstractItemView.SingleSelection)
        self.table.pressed.connect(self.onClicked) 
        self.table.clicked.connect(self.onClickedI) 
        self.table.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        
        data = [
          [4, 9, 2],
          [1, 0, 0],
          [3, 5, 0],
          [3, 3, 2],
          [7, 8, 9],
        ]
        # 传入table作为模型的父对象，方便在flags中获取视图
        self.model = TableModel(data, parent=self.table)
        self.table.setModel(self.model)

        # 为每行最后一列添加按钮
        self.addRowButtons()

        self.setCentralWidget(self.table)

    def addRowButtons(self):
        last_col = self.model.columnCount(QtCore.QModelIndex()) - 1
        row_count = self.model.rowCount(QtCore.QModelIndex())
        for row in range(row_count):
            btn = QtWidgets.QPushButton("操作")
            # 绑定点击事件，传递当前行号
            btn.clicked.connect(lambda checked, r=row: self.onButtonClick(r))
            idx = self.model.index(row, last_col)
            self.table.setIndexWidget(idx, btn)

    def onButtonClick(self, row):
        print(f"点击了第{row+1}行的操作按钮")

    def onClicked(self,r):
        print(r.row()," ",r.column())
    def onClickedI(self,r):
        print(r.row()," ",r.column())


app=QtWidgets.QApplication(sys.argv)
window=MainWindow()
window.show()
app.exec_()

关键说明 #

• setIndexWidget是QTableView的原生方法，与模型类型无关，只要能获取正确的模型索引即可使用，这是核心实现点。
• 模型的flags方法用于控制单元格交互特性，精准实现选中行不可编辑的需求。
• 绑定按钮点击事件时，需将行号作为lambda默认参数传入，避免所有按钮都绑定到最后一行的行号。

内容的提问来源于stack exchange，提问作者Santhosh Dhaipule Chandrakanth