如何在QTableWidget中自定义排序行为

时间:2018-01-29 07:25:45

标签: python sorting pyqt pyqt5 qtablewidget

如何更改列排序行为,而不是按字母顺序排序:

  

1,1.1.1,2,22.2,3等

它将按数字排序,如下所示:

  

1,2,3,11.1,22.2等

我只需要在特定的列而不是整个表上改变排序行为,因为我想保持名称的正常字母顺序。

我还有一个包含文件大小的列。它们以MB和GB表示,我想对它进行排序,以便首先获得800 MB,然后将获得1.4 GB。它还有字母排序的问题。有没有办法可以做到这一点?

1 个答案:

答案 0 :(得分:1)

最简单和最灵活的方法可能是将原始值(用于排序)与格式化值(用于显示)一起存储。然后,您可以继承QTableWidgetItem并重新实现其less than operator,以便在排序时仅比较原始值。这个子类可以有选择地用于需要特殊排序的列。

为了使文件大小正常工作,您需要将原始值存储为字节数,以便可以以相同的方式比较所有格式。

这是一个简单的演示,演示了如何实现这个:

import sys
from PyQt5 import QtCore, QtWidgets

class NumericItem(QtWidgets.QTableWidgetItem):
    def __lt__(self, other):
        return (self.data(QtCore.Qt.UserRole) <
                other.data(QtCore.Qt.UserRole))

class Window(QtWidgets.QTableWidget):
    def __init__(self):
        super(Window, self).__init__(6, 3)
        for column, values in enumerate((
            ('Red', 'Green', 'Yellow', 'Blue', 'White', 'Black'),
            (1, 11.1, 2, 22.2, 3, 17),
            (37885792, 755, 25504, 4805, 3751225472, 14529792),
            )):
            for row, value in enumerate(values):
                if column == 0:
                    item = QtWidgets.QTableWidgetItem(value)
                else:
                    if column == 1:
                        text = str(value)
                    else:
                        text = self.formatSize(value)
                    item = NumericItem(text)
                    item.setData(QtCore.Qt.UserRole, value)
                self.setItem(row, column, item)
        self.setSortingEnabled(True)
        self.sortItems(0, QtCore.Qt.AscendingOrder)

    def formatSize(self, size, precision=2):
        if size < 0:
            return ''
        if size < 1024:
            return '%.0f B' % size
        size /= 1024.0
        if size < 1024:
            return '%.*f KiB' % (precision, size)
        size /= 1024
        if size < 1024:
            return '%.*f MiB' % (precision, size)
        return '%.*f GiB' % (precision, size / 1024)

if __name__ == '__main__':

    app = QtWidgets.QApplication(sys.argv)
    window = Window()
    window.setGeometry(600, 100, 350, 250)
    window.show()
    sys.exit(app.exec_())