PyQt 如何在 QTableView中使用 QComboBox作为代理

在本文中，我们将介绍如何使用 PyQt 中的 QComboBox 作为 QTableView 的代理。QComboBox 是一个下拉列表框，常用于在一个有限的选项列表中进行选择。而 QTableView 是 PyQt 中的一个表格视图控件，可以显示和编辑数据表格。

阅读更多：PyQt 教程

QComboBox 作为 QTableView 的代理

在 QTableView 中使用 QComboBox 作为代理可以实现在表格中的某个单元格中显示下拉列表的效果。这样用户可以从下拉列表中选择一个选项来填写相应的单元格。

使用 QComboBox 作为 QTableView 的代理，需要使用到 QItemDelegate 类。QItemDelegate 是一个代理基类，用于自定义表格的绘制和编辑。

在开始之前，我们需要先创建一个 QTableView 控件，并加载数据。下面是一个简单的例子：

import sys
from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QVBoxLayout, QWidget
from PyQt5.QtCore import QAbstractTableModel, Qt

class MyTableModel(QAbstractTableModel):
    def __init__(self, data):
        super().__init__()
        self.data = data

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

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

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

app = QApplication(sys.argv)
window = QMainWindow()
widget = QWidget()
layout = QVBoxLayout()

data = [['Apple', 'Red', 'Fruit'],
        ['Banana', 'Yellow', 'Fruit'],
        ['Carrot', 'Orange', 'Vegetable'],
        ['Tomato', 'Red', 'Vegetable']]

model = MyTableModel(data)
table = QTableView()
table.setModel(model)

layout.addWidget(table)
widget.setLayout(layout)
window.setCentralWidget(widget)
window.show()
sys.exit(app.exec_())

上面的代码创建了一个简单的 QTableView，并通过 QAbstractTableModel 类加载了一些数据。现在我们可以在表格中显示这些数据。

在下一步中，我们将使用 QComboBox 作为代理来显示第二列的数据。

使用 QComboBox 作为代理

我们首先需要定义一个新的类，继承自 QItemDelegate，用于创建 QComboBox 的实例，并实现一些方法来设置该 QComboBox 的选项。

from PyQt5.QtWidgets import QItemDelegate, QComboBox

class ComboBoxDelegate(QItemDelegate):
    def __init__(self, parent=None):
        super().__init__(parent)

    def createEditor(self, parent, option, index):
        editor = QComboBox(parent)
        editor.addItems(['Red', 'Yellow', 'Green'])
        return editor

    def setEditorData(self, editor, index):
        text = index.model().data(index, Qt.DisplayRole)
        index = editor.findText(text, Qt.MatchFixedString)
        editor.setCurrentIndex(index)

    def setModelData(self, editor, model, index):
        model.setData(index, editor.currentText(), Qt.DisplayRole)

    def updateEditorGeometry(self, editor, option, index):
        editor.setGeometry(option.rect)

上面的代码定义了一个 ComboBoxDelegate 类，该类重写了 QItemDelegate 类的几个方法。

• createEditor: 创建一个 QComboBox 的实例，并设置选项列表为红色、黄色和绿色。
• setEditorData: 设置 QComboBox 的当前选中项为单元格中的值。
• setModelData: 将 QComboBox 中的选中项的值设置到数据模型中。
• updateEditorGeometry: 更新 QComboBox 的位置。

接下来，我们需要将 ComboBoxDelegate 应用到表格的第二列中：

table.setItemDelegateForColumn(1, ComboBoxDelegate())

将上面的代码添加到之前的例子中，然后运行程序，我们可以看到表格的第二列中的单元格显示为 QComboBox，可以通过下拉列表选择相应的选项。

总结

在本文中，我们介绍了如何使用 PyQt 中的 QComboBox 作为 QTableView 的代理。通过使用 QItemDelegate 类和自定义的 ComboBoxDelegate 类，我们可以在表格中的某个单元格中显示一个下拉列表，方便用户选择选项。

使用 QComboBox 作为代理可以提供更多的选项，并且避免了用户输入错误的可能性。这在处理有限的选项列表时非常有用，例如选择颜色、数据类型等等。希望这篇文章对你理解如何在 PyQt 中使用 QComboBox 作为代理有所帮助。