天天看点

PySide6:用树形控件显示数据

如果要显示排列在树中的数据,请使用QTreeWidget。

请注意,使用QTreeWidget并不是在树中显示信息的唯一路径。您还可以创建数据模型并使用QTreeView显示它,但这不在本教程的范围内。

提示:

  1. 为这个应用导入​

    ​QTreeWidget​

    ​ 和​

    ​QTreeWidgetItem​

import sys
from PySide6.QtWidgets import QApplication, QTreeWidget, QTreeWidgetItem      
  1. 定义具有项目结构的字典,以将信息显示为树,其中包含属于每个项目的文件。
data = {"Project A": ["file_a.py", "file_a.txt", "something.xls"],
        "Project B": ["file_b.csv", "photo.jpg"],
        "Project C": []}      
  1. 初始化QApplication单例:
app = QApplication()      
  1. 将QTreeWidget配置为两列,一列用于项目目录中文件的项名称,另一列用于项类型信息。您可以使用setHeaderLabels设置列名,如下所述:
tree = QTreeWidget()
tree.setColumnCount(2)
tree.setHeaderLabels(["Name", "Type"])      
  1. 迭代数据结构,创建QTreeWidgetItem元素,并向每个父级添加相应的子级。我们还提取了仅用于文件的扩展名,并将其添加到第二列中。在构造函数中,您可以看到每个元素(QTreeWidgetItem)都被添加到树(QTree Widget)的不同列中。
items = []
for key, values in data.items():
    item = QTreeWidgetItem([key])
    for value in values:
        ext = value.split(".")[-1].upper()
        child = QTreeWidgetItem([value, ext])
        item.addChild(child)
    items.append(item)

tree.insertTopLevelItems(0, items)      
  1. 显示树并执行QApplication。
tree.show()
sys.exit(app.exec())