QTreeView setModel model.index (2)_重写qtreemodel-优快云博客

本文链接：https://blog.youkuaiyun.com/wowocpp/article/details/106343971

本文介绍了一个使用PyQt5实现的简单模型视图架构案例，通过StringListModel类，展示了如何自定义QAbstractListModel子类来处理字符串列表，并在QListView和QTableView中显示。代码示例涵盖了模型的创建、数据角色的处理以及头数据的提供。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

代码：

import sys

from PyQt5.QtCore import *

from PyQt5.QtWidgets import *

from PyQt5.QtGui import *


class StringListModel(QAbstractListModel):
    def __init__(self, stringList, parent=None):
        super(StringListModel, self).__init__(parent)
        self.stringList = stringList

    # int rowCount(const QModelIndex &parent = QModelIndex()) const;
    # QVariant data(const QModelIndex &index, int role) const;
    # QVariant headerData(int section, Qt::Orientation orientation,
    #                     int role = Qt::DisplayRole) const;

    def rowCount(self,parent=QModelIndex()):

        return len(self.stringList)

    def data(self,index,role= Qt.DisplayRole):

        if (not index.isValid()):
            return QVariant()

        if (index.row() >= len(self.stringList)):
            return QVariant()

        if (role == Qt.DisplayRole):
            return self.stringList[index.row()]
        else:
            return QVariant()

    def headerData(self, section, orientation, role= Qt.DisplayRole):

        if (role != Qt.DisplayRole):
            return QVariant()
        # print(type(section))
        # print(section)
        if (orientation == Qt.Horizontal):
            return "Column %d" % (section)
        else:
            return "Row %d" % (section)



if __name__ == "__main__":

    app = QApplication(sys.argv)

    qlist = list()
    qlist.append("a")
    qlist.append("b")
    qlist.append("c")
    qlist.append("abcd")

    model = StringListModel(qlist)
    qlistView = QListView()

    qlistView.setModel(model)

    qlistView.show()

    tableView = QTableView()

    tableView.setModel(model)

    tableView.show()

    app.exec_()