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

Question

如何更改列排序行为，而不是像这样按字母顺序排序:

1, 11.1, 2, 22.2, 3, etc.

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

1, 2, 3, 11.1, 22.2, etc

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

我还有一个包含文件大小的列。它们以 MB 和 GB 表示，我想对其进行排序，以便 800 MB 排在前面，1.4 GB 排在后面。它还存在字母排序的问题。有没有一种方法可以做到这一点？

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