PyQt5 在 pandas 的 QTableView 上滚动极其缓慢

Question

我正在使用 QTableView 在 PyQt5 GUI 中创建一个 table。我有来自 pandas 数据框的 35 行和 5 列。 table 的滚动和排序非常慢（以秒为单位）。

我已经在寻找解决方案，但大多数人都无法填充 table。有人建议使用 numpy 数组，但我没有看到任何性能提升。

这是我的代码：

def create_table(dataframe):
    table = QTableView()
    tm = TableModel(dataframe)
    table.setModel(tm)

    table.setSelectionBehavior(QAbstractItemView.SelectRows)
    table.resizeColumnsToContents()
    table.resizeRowsToContents()
    table.setSortingEnabled(True)

    return table



class TableModel(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:
                return str(self._data.values[index.row()][index.column()])
        return None

    def headerData(self, rowcol, orientation, role):
        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
            return self._data.columns[rowcol]
        if orientation == QtCore.Qt.Vertical and role == QtCore.Qt.DisplayRole:
            return self._data.index[rowcol]
        return None

    def flags(self, index):
        flags = super(self.__class__, self).flags(index)
        flags |= QtCore.Qt.ItemIsEditable
        flags |= QtCore.Qt.ItemIsSelectable
        flags |= QtCore.Qt.ItemIsEnabled
        flags |= QtCore.Qt.ItemIsDragEnabled
        flags |= QtCore.Qt.ItemIsDropEnabled
        return flags

    def sort(self, Ncol, order):
        """Sort table by given column number.
        """
        try:
            self.layoutAboutToBeChanged.emit()
            self._data = self._data.sort_values(self._data.columns[Ncol], ascending=not order)
            self.layoutChanged.emit()
        except Exception as e:
            print(e)


table = create_table(dataframe)

我在这里 Slow scrolling with QTableView on other comp 发现了一个问题，用户遇到了类似的问题，he/she 发现 "QTableView is refreshing the items at each scroll / appearance of the window, which is obviously the source of the problem." 但是我不知道我的 table 和那个问题一样。

如何让 table 的滚动和排序更快？问题的根源是什么？

Answer 1

问题出在 rowCount 和数据方法中，因为您没有使用最好的函数。在 rowCount 的情况下，当使用值时，您正在创建一个消耗时间的新数据，在这种情况下使用索引。同样在数据中，你必须使用 iloc():

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

# ...

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

PyQt5 在 pandas 的 QTableView 上滚动极其缓慢

PyQt5 Extremely Slow Scrolling on QTableView with pandas

python

pyqt

qtableview

pandas

pyqt5