PyQt5 修改QTableView中的单元格

在本文中，我们将介绍如何使用PyQt5修改QTableView中的单元格。

阅读更多：PyQt5 教程

QTableView简介

QTableView是PyQt5中的一个控件，它用于显示和编辑二维表格数据。它非常适合用于显示大量数据，并且可以灵活地自定义单元格的样式和编辑功能。

要使用QTableView，首先需要创建一个QStandardItemModel对象，并将其设置为QTableView的模型。然后，可以通过添加行和列来填充模型，并将其数据显示在QTableView中。

下面是一个简单的例子，演示了如何使用QTableView显示一个简单的二维表格：

import sys
from PyQt5 import QtWidgets, QtGui

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

        self.table_view = QtWidgets.QTableView()
        self.setCentralWidget(self.table_view)

        model = QtGui.QStandardItemModel(5, 3)
        self.table_view.setModel(model)

        for row in range(5):
            for col in range(3):
                item = QtGui.QStandardItem(f"Item {row}-{col}")
                model.setItem(row, col, item)

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

上面的代码创建了一个5行3列的表格，并将其显示在一个QMainWindow窗口中。每个单元格都包含一个带有行和列索引的文本。这个例子只是QTableView的一个基本用法，接下来我们将介绍如何修改单元格的内容和样式。

修改单元格内容

要修改QTableView中的单元格内容，我们首先需要获取到QTableView的模型对象，然后通过模型对象修改单元格的数据。

在PyQt5中，可以使用QModelIndex对象来表示QTableView中的一个单元格。QModelIndex包含了一些用于访问和修改单元格数据的方法，如data()和setData()等。

下面是一个例子，演示了如何使用setData()方法修改QTableView中的单元格内容：

# ... 省略了上面的代码 ...

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

        self.table_view = QtWidgets.QTableView()
        self.setCentralWidget(self.table_view)

        model = QtGui.QStandardItemModel(5, 3)
        self.table_view.setModel(model)

        for row in range(5):
            for col in range(3):
                item = QtGui.QStandardItem(f"Item {row}-{col}")
                model.setItem(row, col, item)

        # 修改第0行第0列的单元格内容
        index = model.index(0, 0)
        model.setData(index, "New Item")

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

上面的代码修改了第0行第0列的单元格内容为”New Item”。我们使用了model.index(0, 0)方法获取到要修改的单元格的QModelIndex对象，并使用model.setData()方法将新的内容设置到这个单元格中。

修改单元格样式

除了可以修改单元格的内容外，我们还可以修改QTableView中单元格的样式。在PyQt5中，可以通过自定义QStyledItemDelegate来设置单元格的样式。

下面是一个例子，演示了如何使用QStyledItemDelegate自定义QTableView中单元格的样式：

# ... 省略了上面的代码 ...

class MyDelegate(QtWidgets.QStyledItemDelegate):
    def paint(self, painter, option, index):
        # 获取单元格的文本内容
        text = index.data()

        # 绘制单元格背景
        if option.state & QtWidgets.QStyle.State_Selected:
            painter.fillRect(option.rect, option.palette.highlight())
        else:
            painter.fillRect(option.rect, option.palette.base())

        # 绘制文本
        painter.drawText(option.rect, QtCore.Qt.AlignCenter, text)

    def sizeHint(self, option, index):
        return option.rect.size()

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

        self.table_view = QtWidgets.QTableView()
        self.setCentralWidget(self.table_view)

        model = QtGui.QStandardItemModel(5, 3)
        self.table_view.setModel(model)

        for row in range(5):
            for col in range(3):
                item = QtGui.QStandardItem(f"Item {row}-{col}")
                model.setItem(row, col, item)

        delegate = MyDelegate()
        self.table_view.setItemDelegate(delegate)

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

上面的代码自定义了一个MyDelegate类，继承自QStyledItemDelegate，并重写了paint()方法和sizeHint()方法。在paint()方法中，我们根据单元格的状态绘制背景，然后绘制文本。在sizeHint()方法中，我们返回了单元格的大小。

然后，在MyWindow类的初始化方法中，我们创建了一个MyDelegate对象，并将其设置为QTableView的itemDelegate。这样就可以使用自定义的样式绘制单元格了。