QTableView：如何按行索引（标题索引-1）对TableView中的输入数据进行排序？

Question

我正在与QTableView 合作。我能够根据各个列标题以升序和降序对数据进行排序。一旦我按列标题对数据进行排序，我想再次根据原始索引对其进行排序。我尝试关注this question，但无法修复它。

代码：

import numpy as np
from qtpy import QtCore, QtGui, QtWidgets
from qtpy.QtCore import Qt, QAbstractTableModel
from qtpy.QtWidgets import QApplication, QTableView

data = np.random.random((10, 3))

app = QApplication([''])


class SimpleModel(QAbstractTableModel):

    headers = 'Col 1', 'Col 2', 'Col 3'

    def __init__(self, *args, **kwargs):
        super(SimpleModel, self).__init__(*args, **kwargs)
        table.setSortingEnabled(True)
        self.order = np.arange(data.shape[0])
        self.layoutChanged.emit()

    def rowCount(self, index=None):
        return 10

    def columnCount(self, index=None):
        return 3

    def data(self, index, role):
        if role == Qt.DisplayRole:
            col = index.column()
            row = index.row()
            return str(data[self.order[row], col])

    def sort(self, column, ascending):
        self.order = np.argsort(data[:, column])
        if ascending == Qt.DescendingOrder:
            self.order = self.order[::-1]
        self.layoutChanged.emit()

    def headerData(self, section, orientation, role):
        if role == QtCore.Qt.DisplayRole :
            if orientation == QtCore.Qt.Horizontal :
                return self.headers[section]
            elif orientation == QtCore.Qt.Vertical :
                return section
        return None


table = QTableView()
table.setModel(SimpleModel())
table.show()

app.exec_()

截图：

谢谢。

Answer 1

问题是排序方法对数据进行排序并且转换是不可逆的，以前的解决方案使用代理，这对视觉部分而不是数据进行排序，所以最后在其他解决方案的情况下就足够了以避免重新排序并显示原始数据。

import numpy as np
from qtpy import QtCore, QtGui, QtWidgets

data = np.random.random((10, 3))*10 -9

app = QtWidgets.QApplication([''])


class SimpleModel(QtCore.QAbstractTableModel):

    headers = 'Col 1', 'Col 2', 'Col 3'

    def __init__(self, *args, **kwargs):
        super(SimpleModel, self).__init__(*args, **kwargs)
        table.setSortingEnabled(True)
        self.order = np.arange(data.shape[0])
        self.layoutChanged.emit()

    def rowCount(self, index=None):
        return 10

    def columnCount(self, index=None):
        return 3

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.DisplayRole:
            col = index.column()
            row = index.row()
            return str(data[self.order[row], col])

    def headerData(self, section, orientation, role):
        if role == QtCore.Qt.DisplayRole :
            if orientation == QtCore.Qt.Horizontal :
                return self.headers[section]
            elif orientation == QtCore.Qt.Vertical :
                return section
        return None

class SortProxyModel(QtCore.QSortFilterProxyModel):
    def lessThan(self, left_index, right_index):
        left_var = self.sourceModel().data(left_index)
        right_var = self.sourceModel().data(right_index)
        return float(left_var) < float(right_var)


table = QtWidgets.QTableView()
model = SimpleModel()
proxy = SortProxyModel()
proxy.setSourceModel(model)
table.setModel(proxy)
table.show()
cornerButton = table.findChild(QtWidgets.QAbstractButton)
cornerButton.clicked.connect(lambda: proxy.sort(-1))
app.exec_()