qtableview表头添加按钮

在QTableView中添加表头按钮，可以通过自定义表头视图来实现。步骤如下：

1. 继承QHeaderView类，创建自定义的表头视图类（例如MyHeaderView）。
2. 在自定义的表头视图类中重写paintSection()函数，用于绘制表头的每个部分。
3. 在paintSection()函数中判断绘制的是哪个部分，如果是最后一列，就绘制一个按钮。
4. 在重写的paintSection()函数中，监听鼠标点击事件，当点击按钮时，发射一个信号。
5. 在使用QTableView的地方，使用自定义的表头视图类作为表头视图。
6. 在控制QTableView的类中，连接自定义表头视图的信号到槽函数，实现按钮的功能。

以下是一个简单的例子来说明：

from PyQt5.QtWidgets import QTableView, QApplication, QMainWindow, QHeaderView, QWidget, QVBoxLayout, QPushButton
from PyQt5.QtCore import Qt, QAbstractTableModel, QModelIndex
from PyQt5.QtGui import QPainter

class MyHeaderView(QHeaderView):
    def paintSection(self, painter, rect, logicalIndex):
        # 绘制表头的每个部分
        painter.save()
        painter.drawText(rect, Qt.AlignCenter, self.model().headerData(logicalIndex, Qt.Horizontal))
        painter.restore()

        # 如果是最后一列，绘制一个按钮
        if logicalIndex == self.model().columnCount() - 1:
            painter.save()
            buttonRect = rect.adjusted(10, 5, -10, -5)
            painter.drawRoundedRect(buttonRect, 5, 5)
            painter.drawText(buttonRect, Qt.AlignCenter, "按钮")
            painter.restore()

    def mousePressEvent(self, event):
        # 监听鼠标点击事件
        for logicalIndex in range(self.model().columnCount()):
            rect = self.sectionRect(logicalIndex)
            if rect.contains(event.pos()):
                if logicalIndex == self.model().columnCount() - 1:
                    # 点击了最后一列的按钮
                    self.sectionClicked.emit(logicalIndex)
                    return
        super().mousePressEvent(event)

class MyTableModel(QAbstractTableModel):
    def __init__(self):
        super().__init__()
        self.data = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

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

    def columnCount(self, parent=QModelIndex()):
        return len(self.data[0])

    def data(self, index, role=Qt.DisplayRole):
        if role == Qt.DisplayRole:
            return str(self.data[index.row()][index.column()])
        return None

    def headerData(self, section, orientation, role=Qt.DisplayRole):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal:
                return f"列{section}"
            if orientation == Qt.Vertical:
                return f"行{section+1}"
        return None

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("QTableView表头添加按钮")
        self.resize(300, 200)

        self.tableView = QTableView()
        self.tableView.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        
        # 设置自定义的表头视图
        headerView = MyHeaderView(Qt.Horizontal, self.tableView)
        self.tableView.setHorizontalHeader(headerView)

        # 连接自定义表头视图的信号到槽函数
        headerView.sectionClicked.connect(self.onHeaderButtonClicked)

        # 设置数据模型
        model = MyTableModel()
        self.tableView.setModel(model)

        self.setCentralWidget(self.tableView)

    def onHeaderButtonClicked(self, logicalIndex):
        print(f"点击了第{logicalIndex+1}列的按钮")

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

以上是一个简单的示例，实现了在QTableView的表头中添加了一个按钮。点击按钮时，会打印出对应的列号。根据自己的需要，可以进一步定制按钮的外观和功能。