PyQt5-树控件

from PyQt5 import QtCore, QtGui, QtWidgets  
  
class Ui_MainWindow(object):  
  
    def setupUi(self, MainWindow):  
        MainWindow.setObjectName("MainWindow")  
        MainWindow.resize(469, 280)  
        self.centralwidget = QtWidgets.QWidget(MainWindow)  
        self.centralwidget.setObjectName("centralwidget")  
        self.treeView = QtWidgets.QTreeView(self.centralwidget) # 创建树对象  
        self.treeView.setGeometry(QtCore.QRect(0, 0, 471, 281)) # 设置坐标位置和大小  
        # 设置垂直滚动条为按需显示  
        self.treeView.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)  
        # 设置水平滚动条为按需显示  
        self.treeView.setHorizontalScrollBarPolicy(QtCore.Qt.ScrollBarAsNeeded)  
        # 设置双击或者按下回车键时，使树节点可编辑  
        self.treeView.setEditTriggers(QtWidgets.QAbstractItemView.DoubleClicked|QtWidgets.QAbstractItemView.EditKeyPressed)  
        # 设置树节点为单选  
        self.treeView.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)  
        # 设置选中节点时为整行选中  
        self.treeView.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)  
        self.treeView.setAutoExpandDelay(-1) # 设置自动展开的延时为-1，表示自动展开不可用  
        self.treeView.setItemsExpandable(True)    # 设置是否可以展开项  
        self.treeView.setSortingEnabled(True)        # 设置单击头部可排序  
        self.treeView.setWordWrap(True)         # 设置自动换行  
        self.treeView.setHeaderHidden(False)         # 设置不隐藏头部  
        self.treeView.setExpandsOnDoubleClick(True)    # 设置双击可以展开节点  
        self.treeView.setObjectName("treeView")  
        self.treeView.header().setVisible(True) # 设置显示头部  
        MainWindow.setCentralWidget(self.centralwidget)  
        self.retranslateUi(MainWindow)  
        QtCore.QMetaObject.connectSlotsByName(MainWindow)  
        model =QtWidgets.QDirModel()           # 创建存储文件系统的模型  
        self.treeView.setModel(model)           # 为树控件设置数据模型  
  
    def retranslateUi(self, MainWindow):  
        _translate = QtCore.QCoreApplication.translate  
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))  
  
import sys  
# 程序入口，程序从此处启动PyQt设计的窗体  
if __name__ == '__main__':  
   app = QtWidgets.QApplication(sys.argv)  
   MainWindow = QtWidgets.QMainWindow()       # 创建窗体对象  
   ui = Ui_MainWindow()     # 创建PyQt设计的窗体对象  
   ui.setupUi(MainWindow)     # 调用PyQt窗体的方法对窗体对象进行初始化设置  
   MainWindow.show()         # 显示窗体  
   sys.exit(app.exec_())     # 程序关闭时退出进程

from PyQt5 import QtCore, QtGui, QtWidgets  
  
class Ui_MainWindow(object):  
  
    def setupUi(self, MainWindow):  
        MainWindow.setObjectName("MainWindow")  
        MainWindow.resize(422, 197)  
        self.centralwidget = QtWidgets.QWidget(MainWindow)  
        self.centralwidget.setObjectName("centralwidget")  
        # 创建一个TreeView树视图  
        self.treeView = QtWidgets.QTreeView(self.centralwidget)  
        self.treeView.setGeometry(QtCore.QRect(0, 0, 421, 201))  
        self.treeView.setObjectName("treeView")  
        MainWindow.setCentralWidget(self.centralwidget)  
        self.retranslateUi(MainWindow)  
        QtCore.QMetaObject.connectSlotsByName(MainWindow)  
        model = QtGui.QStandardItemModel()      # 创建数据模型  
        model.setHorizontalHeaderLabels(['年级','班级','姓名','分数'])  
  
        # 姓名列表  
        name=['马云','马化腾','李彦宏','王兴','刘强东','董明珠','张一鸣','任正非','丁磊','程维']  
        score=[65,89,45,68,90,100,99,76,85,73] # 分数列表  
  
        import  random  
        # 设置数据  
        for i in range(0,6):  
            # 一级节点：年级，只设第1列的数据  
            grade = QtGui.QStandardItem(("%s年级")%(i + 1))  
            model.appendRow(grade)          # 一级节点  
            for j in range(0,4):  
                # 二级节点：班级、姓名、分数  
                itemClass = QtGui.QStandardItem(("%s班")%(j+1))  
                itemName = QtGui.QStandardItem(name[random.randrange(10)])  
                itemScore = QtGui.QStandardItem(str(score[random.randrange(10)]))  
                # 将二级节点添加到一级节点上  
                grade.appendRow([QtGui.QStandardItem(""),itemClass,itemName,itemScore])  
        self.treeView.setModel(model)          # 为TreeVIew设置数据模型  
  
    def retranslateUi(self, MainWindow):  
        _translate = QtCore.QCoreApplication.translate  
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))  
  
import sys  
# 程序入口，程序从此处启动PyQt设计的窗体  
if __name__ == '__main__':  
   app = QtWidgets.QApplication(sys.argv)  
   MainWindow = QtWidgets.QMainWindow()      # 创建窗体对象  
   ui = Ui_MainWindow()              # 创建PyQt设计的窗体对象  
   ui.setupUi(MainWindow)            # 调用PyQt窗体的方法对窗体对象进行初始化设置  
   MainWindow.show()                 # 显示窗体  
   sys.exit(app.exec_())             # 程序关闭时退出进程

from PyQt5 import QtCore, QtGui, QtWidgets  
from PyQt5.QtWidgets import QTreeWidgetItem  
  
class Ui_MainWindow(object):  
  
    def setupUi(self, MainWindow):  
        MainWindow.setObjectName("MainWindow")  
        MainWindow.resize(240, 150)  
        self.centralwidget = QtWidgets.QWidget(MainWindow)  
        self.centralwidget.setObjectName("centralwidget")  
        self.treeWidget = QtWidgets.QTreeWidget(self.centralwidget)  
        self.treeWidget.setGeometry(QtCore.QRect(0, 0, 240, 150))  
        self.treeWidget.setObjectName("treeWidget")  
        self.treeWidget.setColumnCount(2)  # 设置树结构中的列数  
        self.treeWidget.setHeaderLabels(['姓名', '职务'])  # 设置列标题名  
        root = QTreeWidgetItem(self.treeWidget)  # 创建节点  
        root.setText(0, '组织结构')  # 设置顶级节点文本  
        # 定义字典，存储树结构中显示的数据  
        dict = {'任正非': '华为董事长', '马云': '阿里巴巴创始人', '马化腾': '腾讯CEO', '李彦宏': '百度CEO',  
                '董明珠': '格力董事长'}  
        for key, value in dict.items():  # 遍历字典  
            child = QTreeWidgetItem(root)  # 创建子节点  
            child.setText(0, key)  # 设置第一列的值  
            child.setText(1, value)  # 设置第二列的值  
        self.treeWidget.addTopLevelItem(root)  # 将创建的树节点添加到树控件中  
        self.treeWidget.expandAll()  # 展开所有树节点  
        MainWindow.setCentralWidget(self.centralwidget)  
        self.retranslateUi(MainWindow)  
        QtCore.QMetaObject.connectSlotsByName(MainWindow)  
  
    def retranslateUi(self, MainWindow):  
        _translate = QtCore.QCoreApplication.translate  
        MainWindow.setWindowTitle(_translate("MainWindow", "MainWindow"))  
  
import sys  
# 程序入口，程序从此处启动PyQt设计的窗体  
if __name__ == '__main__':  
    app = QtWidgets.QApplication(sys.argv)  
    MainWindow = QtWidgets.QMainWindow()  # 创建窗体对象  
    ui = Ui_MainWindow()    # 创建PyQt设计的窗体对象  
    ui.setupUi(MainWindow)  # 调用PyQt窗体的方法对窗体对象进行初始化设置  
    MainWindow.show()       # 显示窗体  
    sys.exit(app.exec_())   # 程序关闭时退出进程

# 为节点设置图标  
if key == '任正非':  
    child.setIcon(0, QtGui.QIcon('图标/01.png'))  
elif key == '马云':  
    child.setIcon(0, QtGui.QIcon('图标/02.png'))  
elif key == '马化腾':  
    child.setIcon(0, QtGui.QIcon('图标/03.png'))  
elif key == '李彦宏':  
    child.setIcon(0, QtGui.QIcon('图标/04.png'))  
elif key == '董明珠':  
    child.setIcon(0, QtGui.QIcon('图标/05.png'))

# 为节点设置复选框，并且选中  
child.setCheckState(0, QtCore.Qt.Checked)

self.treeWidget.setAlternatingRowColors(True)  # 设置隔行变色

def gettreetext(self, index):  
    item = self.treeWidget.currentItem()  # 获取当前选中项  
    # 弹出提示框，显示选中项的文本  
    QtWidgets.QMessageBox.information(MainWindow, '提示', '您选择的是：%s -- %s' % (item.text(0), item.text(1)),QtWidgets.QMessageBox.Ok)

# 为树控件绑定单击信号  
self.treeWidget.clicked.connect(self.gettreetext)