PyQt - 实现 QAbstractTableModel 以在 QTableView 中显示

Question

我想在 PyQt 表中显示一个 pandas 数据框。我在这方面取得了一些进展，但未能正确派生表模型类。对此的任何帮助将不胜感激。

\\* 请注意 此处的 完整示例代码 *

我正在努力生成有效的 QtCore.QAbstractTableModel 派生类。继上一个关于 QItemDelegates 的问题之后，我正在尝试从 Pandas DataFrame 生成一个表模型以插入真实数据。我在 这里 有工作示例代码，但是如果我在 Widget 类 (ln 152) 中用 TableModel2 替换我的 TableModel，我将无法显示表格。

 class TableModel2(QtCore.QAbstractTableModel):
    def __init__(self, parent=None, *args):
        super(TableModel2, self).__init__()
        #QtCore.QAbstractTableModel.__init__(self, parent, *args)
        self.datatable = None
        self.headerdata = None
        self.dataFrame = None
        self.model = QtGui.QStandardItemModel(self)

    def update(self, dataIn):
        print 'Updating Model'
        self.datatable = dataIn
        print 'Datatable : {0}'.format(self.datatable)
        headers = dataIn.columns.values
        header_items = [
                    str(field)
                    for field in headers
        ]
        self.headerdata = header_items
        print 'Headers'
        print self.headerdata

        for i in range(len(dataIn.index.values)):
            for j in range(len(dataIn.columns.values)):
                #self.datatable.setItem(i,j,QtGui.QTableWidgetItem(str(df.iget_value(i, j))))
                self.model.setItem(i,j,QtGui.QStandardItem(str(dataIn.iget_value(i, j))))

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

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

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if not index.isValid():
            return QtCore.QVariant()
        elif role != QtCore.Qt.DisplayRole:
            return QtCore.QVariant()
        #return QtCore.QVariant(self.model.data(index))
            return QtCore.QVariant(self.model.data(index))

    def headerData(self, col, orientation, role):
        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
            return QtCore.QVariant()
        return QtCore.QVariant(self.headerdata[col])

    def setData(self, index, value, role=QtCore.Qt.DisplayRole):
        print "setData", index.row(), index.column(), value

    def flags(self, index):
        if (index.column() == 0):
            return QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled
        else:
            return QtCore.Qt.ItemIsEnabled

我正在尝试创建模型，然后将其添加到视图中，如下所示：

 class Widget(QtGui.QWidget):
    """
    A simple test widget to contain and own the model and table.
    """
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)

        l=QtGui.QVBoxLayout(self)
        cdf = self.get_data_frame()
        self._tm=TableModel(self)
        self._tm.update(cdf)
        self._tv=TableView(self)
        self._tv.setModel(self._tm)
        for row in range(0, self._tm.rowCount()):
            self._tv.openPersistentEditor(self._tm.index(row, 0))
        l.addWidget(self._tv)

    def get_data_frame(self):
        df = pd.DataFrame({'Name':['a','b','c','d'],
        'First':[2.3,5.4,3.1,7.7], 'Last':[23.4,11.2,65.3,88.8], 'Class':[1,1,2,1], 'Valid':[True, True, True, False]})
        return df

感谢您的关注！

注意：编辑 2 我已将 QStandardItemModel 合并到 TableModel2 中。在@mata 的评论后还删除了 dataFrameToQtTable 函数。这越来越近了，但仍然没有用。

原文由 C Mars 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

好的，我已经根据上述建议和 Summerfield 的 Rapid GUI 一书中的一些帮助解决了这个问题。 QAbstractTableModel 中不存在底层模型。只需要覆盖三个函数，数据可以以任何用户定义的格式存储，只要在数据调用中返回即可。

一个非常简单的实现可能是：

 class TableModel(QtCore.QAbstractTableModel):
    def __init__(self, parent=None, *args):
        super(TableModel, self).__init__()
        self.datatable = None

    def update(self, dataIn):
        print 'Updating Model'
        self.datatable = dataIn
        print 'Datatable : {0}'.format(self.datatable)

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

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

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.DisplayRole:
            i = index.row()
            j = index.column()
            return '{0}'.format(self.datatable.iget_value(i, j))
        else:
            return QtCore.QVariant()

    def flags(self, index):
        return QtCore.Qt.ItemIsEnabled

这使您能够在 Qt 视图中查看任何兼容的数据框。

我在 这里 更新了要点

如果您还需要这样做，这应该可以让您快速进行。

原文由 C Mars 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

这可能是你的问题：

 def rowCount(self, parent=QtCore.QModelIndex()):
    if type(self.datatable) == pd.DataFrame:
    ...

def columnCount(self, parent=QtCore.QModelIndex()):
    if (self.datatable) == pd.DataFrame:
    ...

You set your datatable to a QTableWidget in dataFrameToQtTable , so it can’t be a pd.DataFrame , your methods will always return 0.

如果没有类型检查，您会立即发现问题。你真的想默默地忽略你的类型不匹配的所有情况吗（如果它不遵循你期望的相同接口，最好让它引发错误）？ 在大多数情况下，类型检查是不必要 的。

原文由 mata 发布，翻译遵循 CC BY-SA 3.0 许可协议