TableView 小部件仅在排序后填充数据

Question

我试图弄清楚如何使用 QTableView 小部件显示我拥有子集的表。我遇到了一个奇怪的错误，我不确定这是错误还是我的代码有问题。我可以很好地加载数据集。当程序运行并按下数据加载按钮时，QTableView 将显示正确的行数和列名，但大多数（如果不是全部）数据都丢失了。但是，如果您尝试通过单击任何列名称对表进行排序，则会出现数据。当我第一次加载数据集时的一些错误消息说“#我们根据定义只有第 0 个轴”或“#我们已经产生了一个标量？”。

我认为这可能是我使用的 csv 文件的问题，但我能够使用随机数字表轻松重现它。注释掉排序功能仍然导致没有数据显示，但现在排序不能用于使数据出现。我在示例中使用的代码是从答案 here 修改而来的。

导入必要的包和函数。

from PyQt5 import QtCore, QtGui, QtWidgets
import pandas as pd
import numpy as np

class PandasModel(QtCore.QAbstractTableModel): 
    def __init__(self, df = pd.DataFrame(), parent=None): 
        QtCore.QAbstractTableModel.__init__(self, parent=parent)
        self._df = df

    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return QtCore.QVariant()

        if orientation == QtCore.Qt.Horizontal:
            try:
                return self._df.columns.tolist()[section]
            except (IndexError, ):
                return QtCore.QVariant()
        elif orientation == QtCore.Qt.Vertical:
            try:
                # return self.df.index.tolist()
                return self._df.index.tolist()[section]
            except (IndexError, ):
                return QtCore.QVariant()

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return QtCore.QVariant()

        if not index.isValid():
            return QtCore.QVariant()

        return QtCore.QVariant(str(self._df.ix[index.row(), index.column()]))

    def setData(self, index, value, role):
        row = self._df.index[index.row()]
        col = self._df.columns[index.column()]
        if hasattr(value, 'toPyObject'):
            # PyQt4 gets a QVariant
            value = value.toPyObject()
        else:
            # PySide gets an unicode
            dtype = self._df[col].dtype
            if dtype != object:
                value = None if value == '' else dtype.type(value)
        self._df.set_value(row, col, value)
        return True

    def rowCount(self, parent=QtCore.QModelIndex()): 
        return len(self._df.index)

    def columnCount(self, parent=QtCore.QModelIndex()): 
        return len(self._df.columns)

    def sort(self, column, order):
        colname = self._df.columns.tolist()[column]
        self.layoutAboutToBeChanged.emit()
        self._df.sort_values(colname, ascending= order == QtCore.Qt.AscendingOrder, inplace=True)
        self._df.reset_index(inplace=True, drop=True)
        self.layoutChanged.emit()

显示表格并处理子集的应用程序。

class Widget(QtWidgets.QWidget):
    def __init__(self, parent=None):
        QtWidgets.QWidget.__init__(self, parent=None)
        vLayout = QtWidgets.QVBoxLayout(self)
        hLayout = QtWidgets.QHBoxLayout()
        self.loadBtn = QtWidgets.QPushButton("Load Table", self)
        hLayout.addWidget(self.loadBtn)
        vLayout.addLayout(hLayout)
        self.pandasTv = QtWidgets.QTableView(self)
        vLayout.addWidget(self.pandasTv)
        self.loadBtn.clicked.connect(self.loadTable)
        #self.pandasTv.setSortingEnabled(True)

    def loadTable(self):
        dfNum = pd.DataFrame(np.random.randint(0,100,size=(20, 4)), columns=list('ABCD'))
        dfNumSub = dfNum.loc[dfNum['A'].astype(str).str.contains('1')]
        model = PandasModel(dfNumSub)
        self.pandasTv.setModel(model)

if __name__ == "__main__":
    import sys
    app = QtWidgets.QApplication(sys.argv)
    w = Widget()
    w.show()
    sys.exit(app.exec_())

我希望显示所有搜索到的数据，而无需对列进行排序。

编辑：此错误仅在设置数据框的子集时发生。更改代码以显示 dfNum 会导致所有数据都正常显示。

Answer 1

显示数据的不是对数据框本身进行排序，而是在PandasModel.sort() 末尾调用self._df.reset_index()。如果没有这个声明，PandasModel.data() 中对self._df.ix() 的调用似乎会中断。要解决此问题，您可以尝试在 Widget.loadTable 中设置模型之前应用 dfNumSub.reset_index()，或者您可以在 PandasModel.data() 中将 self._df.ix() 更改为 self._df.iloc()，因为无论如何不推荐使用 DataFrame.ix。