问题描述
注意:我是一个完整的初学者。
我正在使用 Pandas 数据框导入我从下表转换的 csv 文件。我需要将 csv 文件加载到 QTreeView 中,但一直无法这样做。
输出格式应该是:
我将不得不在某个时候检索“比例类型”,因此我想以某种方式将其标记为该项目,但我不确定这是否可行。我曾尝试将我的数据转换为 Pandas 数据框并加载;直接用csv加载数据;并将其转换为字典,但运气不佳。
我可以对该数据进行硬编码,但我更喜欢使用 csv,因为我以后可以轻松更改它。
目前正在使用:
model.setHeaderData(0,Qt.Horizontal,"Category")
model.setHeaderData(1,"Sub Category")
model.setHeaderData(2,"Test")
self.ui.tv.setModel(model)
self.SetContent(model)
def SetContent(self,model):
self.ui.tv.setModel(model)
i=0
for k,featuretype in features.items():
parent1 = QStandardItem('{}'.format(k[1]))
for item in featuretype:
child = QStandardItem(item[0])
if len(item[1])>0:
for listitem in item[1]:
gchild=QStandardItem(listitem)
child.appendRow(gchild)
parent1.appendRow(child)
model.setItem(i,parent1)
self.ui.tv.setFirstColumnSpanned(i,self.ui.tv.rootIndex(),True)
i+=1
这仅在值是硬键时有效,例如:
features = {('polyGON','SLPR'): [('ONE WAY',['NO','YES','maybe'],'List',3),('CLASS',['INTERSTATE','PRIMARY','RESIDENTIAL','SECONDARY','SERVICE','STATE HWY','TERTIARY','TRACK','US HWY'],11)]
但它不适用于我从 csv 或数据帧创建的字典对象,并且出现“字符串索引超出范围错误”。
我也找到了这段代码,那太好了。但它只给我重复和父母。
reader = csv.reader(f)
for row in reader:
item = QTreeWidgetItem(self.ui.tv,row)
print(row)
解决方法
下面是一个演示脚本,它应该可以满足您的大部分要求。它不能产生与第二个屏幕截图相同的布局,但结构是相同的。 csv 文件转换为嵌套的 dicts/lists,可以保存到 json 文件中。也可以直接加载json文件。我假设您的 csv 文件如下所示:
"Test Category","Sub Category","Test Type","Scale Type"
"Premorbid Func.","SIMPLE","Actual","Scale"
"Premorbid Func.","Predicted","COMPL Montanist","TEST","Scale"
"Intellect","WAIS-IV","T Score"
"Intellect","VCI","Similarities","Vocabulary","T Score"
"Attention","TOVA","RT","Scale"
"Attention","RTV","DV","T","T Score"
这里是树视图的样子:
演示脚本:
import sys,os,csv,json
from collections import defaultdict
from PyQt5 import QtCore,QtGui,QtWidgets
class Window(QtWidgets.QWidget):
def __init__(self):
super().__init__()
self.buttonLoad = QtWidgets.QPushButton('Load Data')
self.buttonLoad.clicked.connect(self.handleProcessData)
self.buttonSave = QtWidgets.QPushButton('Save Data')
self.buttonSave.clicked.connect(self.handleSaveData)
self.buttonSave.setEnabled(False)
self.tree = QtWidgets.QTreeView()
layout = QtWidgets.QGridLayout(self)
layout.addWidget(self.tree,1,2)
layout.addWidget(self.buttonLoad,0)
layout.addWidget(self.buttonSave,1)
self.data = None
def loadData(self):
path,ok = QtWidgets.QFileDialog.getOpenFileName(
self,'Open CSV/JSON','.',filter='Data Files (*.csv *.json)')
if ok:
with open(path) as stream:
if os.path.splitext(path)[1] == '.json':
self.data = json.load(stream)
else:
reader = csv.reader(stream)
# ignore the header
next(reader)
# convert to nested dicts/lists
self.data = defaultdict(lambda: defaultdict(list))
for record in reader:
self.data[record[0]][record[1]].append(record[2:])
def handleProcessData(self):
self.loadData()
if self.data is not None:
model = QtGui.QStandardItemModel(self.tree)
model.setHorizontalHeaderLabels(('Category','Type','Scale'))
self.tree.setModel(model)
self.tree.setColumnWidth(0,200)
root = self.tree.rootIndex()
for row,(text,values) in enumerate(self.data.items()):
category = QtGui.QStandardItem(text)
model.appendRow(category)
self.tree.setFirstColumnSpanned(row,root,True)
for row,values) in enumerate(values.items()):
subcategory = QtGui.QStandardItem(text)
for value in values:
subcategory.appendRow([
QtGui.QStandardItem(),QtGui.QStandardItem(value[0]),QtGui.QStandardItem(value[1]),])
category.appendRow(subcategory)
self.tree.setFirstColumnSpanned(
row,category.index(),True)
self.tree.expandAll()
self.buttonSave.setEnabled(True)
def handleSaveData(self):
path,ok = QtWidgets.QFileDialog.getSaveFileName(
self,'Save JSON',filter='JSON Files (*.json)')
if ok:
with open(path,'w') as stream:
json.dump(self.data,stream,indent=2)
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
window = Window()
window.setWindowTitle('Test')
window.setGeometry(600,100,540,480)
window.show()
sys.exit(app.exec_())