我构建了一棵树,如图所示。
我希望树能够对每一列进行排序 这适用于井名称列和深度类型列(我可能有不同的深度类型,称为“深度”、“MD”、“时间”...),但它不适用于任何其他列。我为每列起诉的项目类型没有区别(据我所知):从字符串构建的简单 Qstandard 项目。
树由嵌套字典和列表填充,格式如下: {好吧:{深度类型:{日志类型:[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())
sort()
函数。当设置 setSortingEnabled()
并调用 sortByColumn()
时(可能通过单击标题部分),视图将调用该函数。
像 QStandardItemModel 这样的完全实现的模型实现了
sort()
每个模型都以自己的方式运行。
在继续解释之前,了解 Qt 项模型的父/子关系如何工作非常重要。
从技术上讲,每个项目都可以是具有自己子项的父项,并且这些子项被映射为table:因此,每个父项可能都有自己的行数和列数,这些行数和列数与模型的行数和列数不匹配(根);不同的行数很明显(“有多少个孩子”),不同的列数不太直观,但也很重要。请注意,QTreeWidget 不会发生这种情况,因为它假设所有子级的列计数都是持久的。
rowCount()
查询模型的 columnCount()
和 rootIndex()
,默认情况下是一个 invalid 索引,映射到模型的 root。
QTreeView 也对第一列的子项执行此操作:实际上,每个父项都将其子项显示为“嵌套表”,第一列中的子项能够拥有自己的更多子项。
现在,QStandardItemModel 的问题和特点是只有显式创建的项目才实际存在。具体来说,子行可以包含比 setColumnCount()
中声明的
少的项目。
这正是问题的原因:当您调用 appendRow()
时,您
不为所有列创建项目。通过以下事实也可以清楚地看出这一点:悬停或单击最后一个“有效”(现有)子项目的右侧不会执行任何操作:那里根本没有项目。请注意,只有一个例外,即顶级项目,原因是它们是 root 索引的子级,然后该索引将
columnCount()
基于与 setColumnCount()
使用的值。
QStandardItemModel 的
sort()
实现仅递归地对 do 在其父级中具有有效项目的列进行排序:例如,如果您有一个具有 4 列的模型,并且父级(不是顶级)仅具有三个有效项目,其子项的第四列将不会被排序。
有两种可能的方法可以解决这个问题。
第一个是为所有列正确创建 all 项,即使它们是空的,这可以在调用
while
之前使用基本的 appendRow()
循环轻松实现:
...
while len(itemList) < model.columnCount():
itemList.append(QStandardItem())
parent.appendRow(itemList)
...
上述必须在所有调用
appendRow()
之前完成。
另一种更合适的方法,也是首选方法,是使用 QSortFilterProxyModel。
WellModel = QStandardItemModel()
WellModel.setColumnCount(6)
WellModel.setHorizontalHeaderLabels(['Well Name','Depth type','Log Type','Log Name','Role','Log unique name'])
proxy = QSortFilterProxyModel()
proxy.setSourceModel(WellModel)
TreeViewLogSelection.setModel(proxy)
QSortFilterProxyModel 不关心 QStandardItemModel 的奇怪实现及其“不存在的项目”,因为它自己实现
sort()
,即使源模型在任何父级别的列计数不一致,排序也将起作用。
使用代理模型的另一个重要好处是它保持原始模型(包括其原始排序)不变。
注意: 1. 只有类和常量才应该有大写的名称,并且你真的不应该对变量使用这样的语法;运算符周围也应使用空格,以提高可读性;请参阅官方的Python代码风格指南,但这些是大多数编程语言的常见约定; 2.
setSortingEnabled()
不应在可能的递归函数中调用。