可编辑的 QTableView 和 Pandas 无法正常工作

Posted 2023-03-29

【中文标题】可编辑的 QTableView 和 Pandas 无法正常工作

【英文标题】：Editable QTableView and Pandas do not work properly

【发布时间】：2017-06-25 05:34:23

【问题描述】：

我正在尝试获取一段独立的示例代码，用于在获取可编辑单元格视图的同时使用 pandas 和 QTableView。

为此，我正在跟进之前的讨论： Pandas df in editable QTableView: remove check boxes

虽然其他讨论中的答案和建议的修改有助于摆脱复选框，但那里讨论的代码仍然不适合我（python 2.7）。

当我使用下面的代码修改单元格时，单元格中显示的内容是：PtQt4.PtCore.QtVariant object at ...

我使用的包版本是：

熊猫：0.20.2 Pyside 1.2.4 Qt 版本：4.8.4 SIP版本：4.14.4 PyQt 版本：4.10

import sys
from PyQt4 import QtCore, QtGui
import pandas as pd
Qt = QtCore.Qt

    class PandasModelEditable(QtCore.QAbstractTableModel):
    def __init__(self, data, parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)
        self._data = data

    def rowCount(self, parent=None):
        return len(self._data.values)

    def columnCount(self, parent=None):
        return self._data.columns.size

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if index.isValid():
            if role == QtCore.Qt.DisplayRole or role == QtCore.Qt.EditRole:
                return unicode(self._data.iloc[index.row(), index.column()])
        return None

    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return None
        if orientation == QtCore.Qt.Horizontal:
            try:
                return '%s' % unicode(self._data.columns.tolist()[section])
            except (IndexError,):
                return unicode()
        elif orientation == QtCore.Qt.Vertical:
            try:
                return '%s' % unicode(self._data.index.tolist()[section])
            except (IndexError,):
                return unicode()

    def flags(self, index):
        return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | \
               QtCore.Qt.ItemIsEditable

    def setData(self, index, value, role=QtCore.Qt.EditRole):
        if index.isValid():
            self._data.iloc[index.row(), index.column()] = value
            if self.data(index, QtCore.Qt.DisplayRole) == value:
                self.dataChanged.emit(index, index)
                return True
        return False


if __name__ == '__main__':
    application = QtGui.QApplication(sys.argv)
    view = QtGui.QTableView()
    df = pd.DataFrame([[1, 2, 3], [4, 5, 6]], columns=['a', 'b', 'c'], index=['x', 'y'])

    model = PandasModelEditable(df)
    view.setModel(model)

    view.show()
    sys.exit(application.exec_())

【参考方案1】：

直接的问题是通过将未转换的QVariant 对象传递给底层数据库引起的。最简单的解决方法是将其转换为 python 对象，如下所示：

self._data.iloc[index.row(), index.column()] = value.toPyObject()

但是，这并不能真正解决代码中最基本的问题，即您使用的是旧版本的 Python 和 PyQt。 Qt 不再正式支持 Qt4，不久之后 Python 和 Python2 也会如此。严格来说，PyQt4 已经是过时的遗留代码 - 所以你不应该将它用于新项目，除非你有一个真的这样做的充分理由（例如向后兼容性） .

如果可以，我强烈建议您尽快将代码移植到 Python3/PyQt5，因为从中长期来看，这将为您省去很多麻烦。但是，如果由于某种原因无法执行此操作，并且希望继续使用 Python2/PyQt4，则可以通过在程序开头添加以下内容来获得与 PySide 相同的行为：

import sip
sip.setapi('QString', 2)
sip.setapi('QVariant', 2)
from PyQt4 import QtCore, QtGui

完成此操作后，PyQt 会自动将所有QString 和QVariant 对象转换为普通的python 数据类型，因此您将永远不需要进行任何显式转换（即您可以删除所有那些unicode() 和toPyObject()在您的代码中调用）。

或者，您也可以将 Python3 与 PyQt4 一起使用，默认情况下它与 PySide 具有相同的行为（因此不需要 setapi 的东西）。

【参考方案2】：

当我切换到 PySide 而不是 PyQt4 时，它似乎可以工作：

import sys
from PySide import QtCore, QtGui
import pandas as pd
Qt = QtCore.Qt

    class PandasModelEditable(QtCore.QAbstractTableModel):
    def __init__(self, data, parent=None):
        QtCore.QAbstractTableModel.__init__(self, parent)
        self._data = data

    def rowCount(self, parent=None):
        return len(self._data.values)

    def columnCount(self, parent=None):
        return self._data.columns.size

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if index.isValid():
            if role == QtCore.Qt.DisplayRole or role == QtCore.Qt.EditRole:
                return unicode(self._data.iloc[index.row(), index.column()])
        return None

    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return None
        if orientation == QtCore.Qt.Horizontal:
            try:
                return '%s' % unicode(self._data.columns.tolist()[section])
            except (IndexError,):
                return unicode()
        elif orientation == QtCore.Qt.Vertical:
            try:
                return '%s' % unicode(self._data.index.tolist()[section])
            except (IndexError,):
                return unicode()

    def flags(self, index):
        return QtCore.Qt.ItemIsEnabled | QtCore.Qt.ItemIsSelectable | \
               QtCore.Qt.ItemIsEditable

    def setData(self, index, value, role=QtCore.Qt.EditRole):
        if index.isValid():
            self._data.iloc[index.row(), index.column()] = value
            if self.data(index, QtCore.Qt.DisplayRole) == value:
                self.dataChanged.emit(index, index)
                return True
        return False


if __name__ == '__main__':
    application = QtGui.QApplication(sys.argv)
    view = QtGui.QTableView()
    df = pd.DataFrame([[1, 2, 3], [4, 5, 6]], columns=['a', 'b', 'c'], index=['x', 'y'])

    model = PandasModelEditable(df)
    view.setModel(model)

    view.show()
    sys.exit(application.exec_())