首页 > 解决方案 > QTableView:如何按行索引(标题索引-1)对TableView中的输入数据进行排序?

问题描述

我正在与QTableView. 我能够根据各个列标题以升序和降序对数据进行排序。一旦我按列标题对数据进行排序,我想再次根据原始索引对其进行排序。我尝试关注这个问题,但无法解决。

代码:

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_()

截屏:

在此处输入图像描述

谢谢你。

标签: pythonpyqtpyqt5qtableviewqtablewidget

解决方案


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

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_()

推荐阅读