PyQt5?QTreeWidget樹形結構遞歸遍歷當前所有節點的實現方法是什么

這篇文章主要介紹“PyQt5 QTreeWidget樹形結構遞歸遍歷當前所有節點的實現方法是什么”，在日常操作中，相信很多人在PyQt5 QTreeWidget樹形結構遞歸遍歷當前所有節點的實現方法是什么問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”PyQt5 QTreeWidget樹形結構遞歸遍歷當前所有節點的實現方法是什么”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

QTreeWidget類中的常用方法

QTreeWidgetItem類中的常用方法

準備一個小demo

常見小demo
簡介：商品種類和商品的樹形嵌套結構demo
功能：點擊按鍵獲取當前所有被選中的商品(夾帶了一點私貨〃‘▽'〃)
注意：篇幅有限，沒有寫子節點和父節點聯動選中，所以在選擇子節點時麻煩自行將父節點選上，不然會跳過。

代碼塊兒：

import sys
from PyQt5.QtWidgets import QTreeWidgetItem, QTreeWidget, QWidget, QVBoxLayout, QPushButton, QApplication
from PyQt5.QtCore import Qt


class Demo(QWidget):
    def __init__(self):
        super().__init__()
        # 實例化一個樹形結構，隱藏了header
        self.tree = QTreeWidget()
        self.tree.setHeaderHidden(True)
        # 頂級分支
        self.tree_main = QTreeWidgetItem(self.tree)
        self.tree_main.setText(0, '商品種類')
        # 設置一些二級分支
        tree_second = ['電子產品', '水果', '日用品', '喜歡的人']
        self.gen_branch(self.tree_main, tree_second)
        # 設置一些三級分支
        tree_fruit = ['蘋果', '香蕉', '梨']
        tree_daily_use = ['紙巾', '毛巾']
        tree_lovers = ['迪迪1號', '迪迪2號']
        # child(1) 意思是分支的第1個節點, 序號從0算起
        self.gen_branch(self.tree_main.child(1), tree_fruit)
        self.gen_branch(self.tree_main.child(2), tree_daily_use)
        self.gen_branch(self.tree_main.child(3), tree_lovers)
        # 一個按鈕
        self.pushButton = QPushButton('選好了')
        # 顯示出來
        self.qvl = QVBoxLayout()
        self.qvl.addWidget(self.tree)
        self.qvl.addWidget(self.pushButton)
        self.setLayout(self.qvl)

        # 綁定一下槽函數，傳入主要的分支節點
        self.pushButton.clicked.connect(lambda: self.get_checked(self.tree_main))

    @staticmethod
    def gen_branch(node: QTreeWidgetItem, texts: list):
        """ 給定某個節點和列表 在該節點生成列表內分支"""
        for text in texts:
            item = QTreeWidgetItem()
            item.setText(0, text)
            item.setCheckState(0, Qt.Unchecked)
            node.addChild(item)

    def get_checked(self, node: QTreeWidgetItem)->list:
        """ 得到當前節點選中的所有分支， 返回一個 list """
        temp_list = []
        # 此處看下方注釋 1
        for item in node.takeChildren():
            # 判斷是否選中
            if item.checkState(0) == Qt.Checked:
                temp_list.append(item.text(0))
                # 判斷是否還有子分支
                if item.childCount():
                    temp_list.extend(self.get_checked(item))
            node.addChild(item)
        print(temp_list)
        return temp_list


if __name__ == '__main__':
    app = QApplication(sys.argv)
    win = Demo()
    win.show()
    sys.exit(app.exec_())

注釋01：在這個函數中，我傳入了一個 node 節點，takeChildren() 這個方法會將該node節點的所有一級子分支拿出來(刪除)，并返回節點的所有一級分支的列表，如下所示。該方法只能返回一級的節點信息，利用 childCount() 來判斷是否有子分支，有則遞歸，一直到最底部的節點。因為在獲取的時候 takeChildren() 刪除了所有節點，所以在操作結束后重新加入到 node 節點中

[
<PyQt5.QtWidgets.QTreeWidgetItem object at 0x0000000008464708>, 
<PyQt5.QtWidgets.QTreeWidgetItem object at 0x0000000008464798>, 
]

這樣的做法有什么好處和壞處？

最大的好處無疑是不需要創建額外的變量去存儲子節點的信息，子節點的信息和順序都是實時獲取的而非前期就定好了的。壞處，我設想這個方法用多了可能會存在節點順序改變的情況。比如“蘋果、香蕉”變成了“香蕉、蘋果”，目前未出現。

對比網上的方法

有一個關于 QTreeWidgetItemIterator 的辦法，這是Qt中自帶的遍歷器，大概如下

item = QtWidgets.QTreeWidgetItemIterator(self.treeWidget)，

用 item.value() 來定位到一個節點，item.value() 的實例就是上文列表中的那種對象，個人感覺差不太多。

還有一種比較暴力做法。在生成子節點的時候將所有子節點放到當前類的作用域中，也就是作為屬性存在。

self.item1 = QTreeWidgetItem()

或是生成的時候保存在一個定義在作用域的列表中，這么做有一個壞處，節點的信息都是提前定好了的。但實際上遇到的情況更多應該是未知的。

self.item_list = []
self.item_list.append([... ... ])

到此，關于“PyQt5 QTreeWidget樹形結構遞歸遍歷當前所有節點的實現方法是什么”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！