我建了一棵树,如图所示。
我希望树能够对每个列进行排序,这对“井名”列和“深度类型”列有用(我可能有不同的深度类型,称为“深度”、“MD”、“时间”等),但它对任何其他列都不起作用。据我所知,我为每列使用的项目类型没有区别:一个由字符串构建的简单 Qstandard 项目。
该树由嵌套字典和列表填充,形式如下:{well:{depthtype:{logtype:[log1,log2,log3]}}}
使用以下代码
def fillWellTree(self, parent, dico, depth=0):
if isinstance(dico, dict):
for key, value in dico.items():
item1=QStandardItem(str(key))
item1.setEditable(False)
itemList=[item1]
if depth==0:
itemList[0].setIcon(QIcon("./icon/IconWell.png"))
if isinstance(value, dict):
for i in range(depth):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemList.insert(0,emptyItem)
parent.appendRow(itemList)
self.fillWellTree(itemList[0], value, depth+1)
elif isinstance(value, list):
for i in range(depth):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemList.insert(0,emptyItem)
parent.appendRow(itemList)
for val in value:
item_i=QStandardItem(str(val))
item_i.setEditable(False)
itemLogList=[item_i]
for i in range(depth+1):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemLogList.insert(0,emptyItem)
itemList[0].appendRow(itemLogList)
我想知道排序是否不起作用,因为最后 2 个填充列(日志类型和日志名称)中包含数据的行的父项有空字符串,所以我用随机字符串替换了空字符串,但它没有改变任何东西谢谢你的帮助和建议
这里编辑是一个工作脚本:
import sys
from PySide6.QtWidgets import QApplication, QTreeView
from PySide6.QtGui import QStandardItem, QStandardItemModel
app = QApplication(sys.argv)
TreeViewLogSelection=QTreeView()
WellModel = QStandardItemModel()
WellModel.setColumnCount(6)
TreeViewLogSelection.setModel(WellModel)
TreeViewLogSelection.model().setHorizontalHeaderLabels(['Well Name','Depth type','Log Type','Log Name','Role','Log unique name'])
def fillWellTree(parent, dico, depth=0):
if isinstance(dico, dict):
for key, value in dico.items():
item1=QStandardItem(str(key))
item1.setEditable(False)
itemList=[item1]
if isinstance(value, dict):
for i in range(depth):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemList.insert(0,emptyItem)
parent.appendRow(itemList)
fillWellTree(itemList[0], value, depth+1)
elif isinstance(value, list):
for i in range(depth):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemList.insert(0,emptyItem)
parent.appendRow(itemList)
for val in value:
item_i=QStandardItem(str(val))
item_i.setEditable(False)
itemLogList=[item_i]
for i in range(depth+1):
emptyItem=QStandardItem("")
emptyItem.setEditable(False)
itemLogList.insert(0,emptyItem)
itemList[0].appendRow(itemLogList)
TreeViewLogSelection.setSortingEnabled(True)
dico={'Well-6': {'Depth': {'Sonic': ['DT', 'DTS', 'DTST', 'VPVS', 'DT_REG', 'Smoothing (DT_REG)', 'DT_FINAL'], 'Shallow resistivity': ['LLS', 'MSFL'], 'Bulk density': ['RHOB_RAW', 'RHOB_Predict_RFA', 'RHOB_REG', 'RHOB_DESPIKED', 'RHOB_DESPIKE_REG', 'Smoothing (RHOB_DESPIKE_REG)', 'RHOB_FINAL', 'RHOB_Predict_RF'], 'Shear slowness': ['DTS_Predict_RF', 'DTS_Predict_NN'], 'Deviation': ['DX', 'DY']}, 'MD': {'Sonic': ['DT_Predict_RF','DTS_predict']}}, 'DRILL-1': {'Depth': {'Bit size': ['BS'], 'Caliper': ['CALI'], 'Gamma ray': ['GR'], 'Neutron': ['NPHI'], 'Photoelectric factor': ['PE'], 'Spontaneous potential': ['SP'], 'Shallow resistivity': ['LLS'], 'Deep resistivity': ['LLD'], 'Sonic': ['DT', 'DTS','DTC'], 'Bulk density': ['RHOB'], 'Anonymous': ['WELLTOPS'], 'Badhole indicator': ['Bad_Hole']}}, 'WELLI-1': {'Depth': {'Sonic': ['DT', 'DTS','DTC'], 'Gamma ray': ['GR', 'GR2'], 'Shallow resistivity': ['LLS', 'MSFL'], 'Bulk density': ['RHOB_RAW', 'RHOB_Predict_RF']}, 'MD': {'Sonic': ['DT_Predict_RF', 'DT_Merged', 'DT_FINAL'], 'Impedance': ['AI','IP']}}}
fillWellTree(WellModel.invisibleRootItem(),dico)
TreeViewLogSelection.show()
sys.exit(app.exec())
当需要对项目进行排序时,Qt 项目模型会使用该函数。当设置并调用时(可能是通过单击标题部分),
sort()
视图会调用该函数。setSortingEnabled()
sortByColumn()
完全实现的模型,如 QStandardItemModel,
sort()
以自己的方式实现每个功能。在继续解释之前,了解 Qt 项目模型的父/子关系如何运作非常重要。
从技术上讲,每个项目都可以是具有自己的子项的父项,并且这些子项被映射为一个表:因此,每个父项可能都有自己的行数和列数,这些行数和列数与模型(根)的行数和列数不匹配;不同的行数很明显(“有多少个子项”),不同的列数不太直观,但也很重要。请注意,QTreeWidget 不会发生这种情况,因为它假定列数对于所有子项都是持久的。
视图通常根据模型的来查询模型的
rowCount()
并columnCount()
基于其rootIndex()
(默认情况下是无效索引)映射到模型的根。QTreeView 也对第一列的子项执行此操作:实际上,每个父项都将其子项显示为“嵌套表”,并且第一列的子项可以自行拥有进一步的子项。
现在,QStandardItemModel 的问题和特性是只有明确创建的项目才会实际存在。具体来说,子行的项目数可以少于中声明的项目数
setColumnCount()
。这正是导致问题的原因:调用 时,您并未
appendRow()
为所有列创建项目。这一点也显而易见,因为将鼠标悬停或单击最后一个“有效”(现有)子项的右侧不会执行任何操作:那里根本没有项目。请注意,只有一个例外,即顶级项目,原因是它们是根索引的子项,然后基于columnCount()
所使用的值setColumnCount()
。sort()
QStandardItemModel 的实现仅以递归方式对父级中具有有效项目的列进行排序:例如,如果您有一个有 4 列的模型,并且父级(不是顶层)只有三个有效项目,则其子级的第四列将不会被排序。有两种可能的方法可以解决这个问题。
首先是正确地创建所有列的所有项目,即使它们是空的,这可以
while
在调用之前通过基本循环轻松实现appendRow()
:上述操作必须在所有调用之前完成
appendRow()
。另一种更合适的方法(也是首选方法)是使用QSortFilterProxyModel。
QSortFilterProxyModel 不关心 QStandardItemModel 的奇怪实现及其“不存在的项目”,因为它可以
sort()
自行实现,并且即使源模型在任何父级别具有不一致的列数,排序也会有效。使用代理模型的另一个重要好处是它保持原始模型(包括其原始排序)不变。
注意:1. 只有类和常量才应该使用大写名称,并且您不应该对变量使用这样的语法;还应该在运算符周围使用空格,以提高可读性;请参阅官方的《Python 代码样式指南》,但这些是大多数编程语言的通用惯例;2.
setSortingEnabled()
不应在可能的递归函数中调用。