Question

我想按QTreeWidget列对alpha_numeric进行排序。

更新

关注ekhumoro's example后，我重新实施了QTreeWidgetItem __lt__方法以使用desired natural sorting。

相关代码：

class TreeWidgetItem(QtGui.QTreeWidgetItem):
    def __lt__(self, other):
        column = self.treeWidget().sortColumn()
        key1 = self.text(column)
        key2 = other.text(column)
        return self.natural_sort_key(key1) < self.natural_sort_key(key2)

    @staticmethod
    def natural_sort_key(key):
        regex = '(\d*\.\d+|\d+)'
        parts = re.split(regex, key)
        return tuple((e if i % 2 == 0 else float(e)) for i, e in enumerate(parts))

更新了完整的工作示例：

import sys
import re
from PySide import QtGui, QtCore

data = [
    {"name": "Apple", "alpha_numeric": "11"},
    {"name": "Apple", "alpha_numeric": "125b"},
    {"name": "Apple", "alpha_numeric": "125a"},
    {"name": "Apple", "alpha_numeric": "3"},
    {"name": "Orange", "alpha_numeric": "11"},
    {"name": "Orange", "alpha_numeric": "125b"},
    {"name": "Orange", "alpha_numeric": "125a"},
    {"name": "Orange", "alpha_numeric": "3"},
]

# re-implement the QTreeWidgetItem
class TreeWidgetItem(QtGui.QTreeWidgetItem):
    def __lt__(self, other):
        column = self.treeWidget().sortColumn()
        key1 = self.text(column)
        key2 = other.text(column)
        return self.natural_sort_key(key1) < self.natural_sort_key(key2)

    @staticmethod
    def natural_sort_key(key):
        regex = '(\d*\.\d+|\d+)'
        parts = re.split(regex, key)
        return tuple((e if i % 2 == 0 else float(e)) for i, e in enumerate(parts))


class MainWindow(QtGui.QMainWindow):
    def __init__(self):
        super(MainWindow, self).__init__()
        self.resize(300, 300)
        self.table = None
        self.ui()

    def ui(self):
        # setup the table
        self.table = QtGui.QTreeWidget()
        self.table.setRootIsDecorated(False)
        self.table.setAlternatingRowColors(True)
        self.table.setSortingEnabled(True)

        # columns and widths
        column_names = ['Name', 'Alpha Numeric']
        self.table.setColumnCount(2)
        self.table.setColumnWidth(0, 150)
        self.table.setColumnWidth(1, 150)
        self.table.setHeaderLabels(column_names)

        # row data
        for row in data:
            column = TreeWidgetItem(self.table)
            column.setSizeHint(0, QtCore.QSize(30, 30))
            column.setText(0, row['name'])
            column.setText(1, row['alpha_numeric'])

        # sort
        self.table.sortByColumn(1, QtCore.Qt.AscendingOrder)
        self.table.sortByColumn(0, QtCore.Qt.AscendingOrder)

        # setup the layout
        self.setCentralWidget(self.table)


if __name__ == '__main__':
    app = QtGui.QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec_())

Answer 1

您可以重新实现表格小部件项目的less-than operator，并使用您喜欢的任何算法来控制排序：

class TableWidgetItem(QtGui.QTableWidgetItem):
    def __lt__(self, other):
        return funky_sort_key(self.text(), other.text())

使用此方法，您甚至不需要按文本排序 - 您还可以按项目的数据（或其他）进行排序。

通过字母数字列对PySide.QtGui.QTreeWidget进行排序

1 个答案: