PyQt5 TableView 点击某一列标题进行排序|极客教程

PyQt5 TableView 点击某一列标题进行排序

导言

在图形界面应用程序的开发中，表格是一个常见的控件，用于展示和处理大量的数据。PyQt5是一个强大的Python库，提供了丰富而灵活的工具来创建图形界面应用程序。其中的QTableView控件是用于显示和编辑表格数据的强大工具。在本文中，我们将探讨如何使用PyQt5中的QTableView控件，并在点击表格列标题时进行排序。

简介

在现代应用程序开发中，数据通常以表格的形式展示。这种展示方式可以帮助用户对数据进行分析和操作。PyQt5提供了QTableView控件，它是一个灵活的表格控件，可以用于显示和编辑数据。通过对QTableView控件的定制和扩展，我们可以实现各种复杂的表格功能。

QTableView控件

QTableView是PyQt5中用于显示和编辑表格数据的核心控件之一。它以表格的形式展示数据，支持用户对表格数据的排序、筛选和编辑等操作。下面的示例演示了如何使用QTableView控件创建一个简单的表格：

from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QVBoxLayout, QWidget
from PyQt5.QtCore import Qt
from PyQt5.QtGui import QStandardItemModel, QStandardItem

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("Table View Example")
        self.resize(400, 300)

        # 创建表格模型
        self.model = QStandardItemModel(4, 2)
        self.model.setHeaderData(0, Qt.Horizontal, "姓名")
        self.model.setHeaderData(1, Qt.Horizontal, "年龄")
        for row in range(4):
            for column in range(2):
                item = QStandardItem(str(row+1) + "行 " + str(column+1) + "列")
                self.model.setItem(row, column, item)

        # 创建表格视图
        self.table_view = QTableView()
        self.table_view.setModel(self.model)

        # 设置表格自适应大小
        self.table_view.resizeColumnsToContents()
        self.table_view.resizeRowsToContents()

        # 创建主窗口布局
        layout = QVBoxLayout()
        layout.addWidget(self.table_view)

        # 创建主窗口中心部件
        central_widget = QWidget()
        central_widget.setLayout(layout)

        # 设置主窗口中心部件
        self.setCentralWidget(central_widget)

if __name__ == "__main__":
    app = QApplication([])
    main_window = MainWindow()
    main_window.show()
    app.exec()

运行以上代码，将会创建一个简单的窗口，并在窗口中显示一个2列4行的表格。表格的列标题分别为“姓名”和“年龄”，而单元格数据为每个单元格所在行列的信息。

点击表格列标题进行排序

在实际应用中，我们通常希望能够根据某一列的值来对表格数据进行排序。PyQt5提供了QSortFilterProxyModel类，可以用于对QTableView控件的数据进行排序和筛选。下面的示例演示了如何通过点击表格列标题来实现对表格列的排序功能：

from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QVBoxLayout, QWidget
from PyQt5.QtCore import Qt, QSortFilterProxyModel
from PyQt5.QtGui import QStandardItemModel, QStandardItem

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.setWindowTitle("Table View Example")
        self.resize(400, 300)

        # 创建表格模型
        self.model = QStandardItemModel(4, 2)
        self.model.setHeaderData(0, Qt.Horizontal, "姓名")
        self.model.setHeaderData(1, Qt.Horizontal, "年龄")
        for row in range(4):
            for column in range(2):
                item = QStandardItem(str(row+1) + "行 " + str(column+1) + "列")
                self.model.setItem(row, column, item)

        # 创建排序代理模型
        self.proxy_model = QSortFilterProxyModel()
        self.proxy_model.setSourceModel(self.model)

        # 创建表格视图
        self.table_view = QTableView()
        self.table_view.setModel(self.proxy_model)

        # 设置表格自适应大小
        self.table_view.resizeColumnsToContents()
        self.table_view.resizeRowsToContents()

        # 连接点击表格列标题的事件
        self.table_view.horizontalHeader().sectionClicked.connect(self.sort_table_by_column)

        # 创建主窗口布局
        layout = QVBoxLayout()
        layout.addWidget(self.table_view)

        # 创建主窗口中心部件
        central_widget = QWidget()
        central_widget.setLayout(layout)

        # 设置主窗口中心部件
        self.setCentralWidget(central_widget)

    def sort_table_by_column(self, column):
        """根据表格列标题进行排序"""

        # 获取当前排序列的排序方式
        current_sort_order = self.proxy_model.sortOrder()

        # 切换排序方式（升序/降序）
        new_sort_order = Qt.AscendingOrder if current_sort_order == Qt.DescendingOrder else Qt.DescendingOrder

        # 设置排序列和排序方式
        self.proxy_model.setSortRole(Qt.DisplayRole)
        self.proxy_model.sort(column, new_sort_order)

if __name__ == "__main__":
    app = QApplication([])
    main_window = MainWindow()
    main_window.show()
    app.exec()

运行以上代码，将会创建一个窗口，并在窗口中显示一个2列4行的表格。当点击表格列标题时，表格数据将会根据该列的值进行排序。如果首次点击列标题，数据将会按升序排序；如果再次点击同一列标题，数据将会按降序排序。