Python PyQt5中TableView行选择颜色|极客教程

Python PyQt5中TableView行选择颜色

在PyQt5中，TableView是一种常用的控件，用于显示和编辑表格数据。在实际开发中，经常需要对TableView中的行进行选择，并且对选中的行进行特殊的颜色显示。本文将详细介绍如何在PyQt5中实现TableView行选择并定义选中行的颜色。

基本概念

在PyQt5中，可以使用QTableView控件来显示表格数据。QTableView是基于Model/View架构的控件，通过继承QAbstractTableModel和QTableView类，可以自定义数据模型和表格视图。

在本文中，我们将使用QStandardItemModel类作为数据模型，用于存储表格数据。同时，我们将创建一个自定义的TableView类，继承自QTableView，用于显示表格数据，并实现行选择颜色的功能。

实现步骤

导入必要的模块：

from PyQt5.QtWidgets import QApplication, QTableView, QHeaderView
from PyQt5.QtGui import QColor, QBrush
from PyQt5.QtCore import Qt, QModelIndex, QAbstractTableModel
import sys

创建自定义的数据模型类CustomTableModel，继承自QAbstractTableModel，并实现必要的方法：

class CustomTableModel(QAbstractTableModel):
    def __init__(self, data, headers):
        super().__init__()
        self._data = data
        self._headers = headers

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

    def columnCount(self, parent):
        return len(self._data[0])

    def data(self, index, role):
        if not index.isValid():
            return None

        if role == Qt.DisplayRole:
            return self._data[index.row()][index.column()]

        return None

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

        return None

创建自定义的TableView类CustomTableView，继承自QTableView，并实现行选择颜色的功能：

class CustomTableView(QTableView):
    def __init__(self, parent=None):
        super().__init__(parent)

        self.setSelectionBehavior(QTableView.SelectRows)
        self.setSelectionMode(QTableView.SingleSelection)
        self.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)
        self.verticalHeader().setVisible(False)

    def currentChanged(self, current, previous):
        if current.isValid():
            self.selectRow(current.row())

    def selectRow(self, row):
        for column in range(self.model().columnCount(QModelIndex())):
            index = self.model().index(row, column)
            if index.isValid():
                self.setRowColor(index, QColor(Qt.blue))

    def setRowColor(self, index, color):
        self.model().setData(index, QBrush(color), Qt.BackgroundRole)

创建应用程序：

if __name__ == '__main__':
    app = QApplication(sys.argv)

    # 模拟表格数据
    data = [
        ["Alice", "25", "Female"],
        ["Bob", "30", "Male"],
        ["Charlie", "35", "Male"],
        ["David", "40", "Male"]
    ]

    headers = ["Name", "Age", "Gender"]

    # 创建数据模型
    model = CustomTableModel(data, headers)

    # 创建TableView
    tableView = CustomTableView()
    tableView.setModel(model)
    tableView.show()

    sys.exit(app.exec_())