极客笔记Python tablewidget 自动排序
简介
在使用Python编写GUI程序时，经常需要使用表格来展示和编辑数据。PyQt库提供了一个非常方便的控件——QTableWidget，用于在图形界面中创建表格。其中，对于需要展示大量数据的表格，常常需要进行排序操作，以方便用户查找和浏览数据。本文将详细介绍如何在QTableWidget中实现自动排序功能。
1. 创建表格
首先，我们需要创建一个基本的QTableWidget实例，并预设好表格的行数和列数。可以使用以下代码创建一个5行3列的表格：
import sys
from PyQt5.QtWidgets import QApplication, QTableWidget, QTableWidgetItem
app = QApplication(sys.argv)
table = QTableWidget()
table.setRowCount(5)
table.setColumnCount(3)
接下来，我们可以在表格中填充数据。可以使用setItem方法来设置每个单元格的内容。例如，以下代码将数据填充到第一行第一列的单元格：
item = QTableWidgetItem("Hello")
table.setItem(0, 0, item)
2. 设置表头
在表格的第一行通常会显示表头，即每一列的名称。可以使用setHorizontalHeaderLabels方法来设置表头。例如，以下代码将表头设置为”姓名”、”年龄”和”性别”：
header_labels = ["姓名", "年龄", "性别"]
table.setHorizontalHeaderLabels(header_labels)
3. 添加数据
如果要一次性添加多条数据到表格中，可以使用setRowCount方法指定表格的行数，并使用循环将数据逐一添加到每个单元格中。例如，以下代码添加三条数据到表格中：
data = [
["小明", "18", "男"],
["小红", "22", "女"],
["小刚", "20", "男"]
]
table.setRowCount(len(data))
for i, row in enumerate(data):
for j, cell in enumerate(row):
item = QTableWidgetItem(cell)
table.setItem(i, j, item)
4. 实现排序功能
要实现自动排序功能，我们需要做以下几步：
4.1. 添加排序按钮
首先，我们需要添加一个排序按钮，如”按姓名排序”按钮。可以使用QPushButton来创建按钮，并使用addWidget方法将按钮添加到表格的右上角。例如，以下代码创建一个”按姓名排序”的按钮：
from PyQt5.QtWidgets import QPushButton, QVBoxLayout, QWidget
# 创建按钮
sort_button = QPushButton("按姓名排序")
# 添加按钮到表格的右上角
widget = QWidget()
layout = QVBoxLayout(widget)
layout.addWidget(sort_button)
layout.addWidget(table)
4.2. 绑定排序按钮的点击事件
接下来，我们需要为排序按钮绑定点击事件的处理函数，以实现点击按钮时进行排序操作。可以使用以下代码为按钮绑定点击事件处理函数：
def sort_by_name():
table.sortByColumn(0)
sort_button.clicked.connect(sort_by_name)
上述代码中，sort_by_name函数将会在点击按钮时被调用，然后调用sortByColumn方法以表格的第一列作为排序列进行排序。你可以通过修改sortByColumn方法的参数来指定不同的列进行排序。
4.3. 创建自定义表格模型
默认情况下，QTableWidget的排序功能是不生效的。为了实现自动排序功能，我们需要创建一个自定义的表格模型，并在该模型中重写排序相关的方法。
from PyQt5.QtCore import Qt, QAbstractTableModel
class CustomTableModel(QAbstractTableModel):
def __init__(self, data, header_labels, parent=None):
super().__init__(parent)
self._data = data
self._header_labels = header_labels
def rowCount(self, parent):
return len(self._data)
def columnCount(self, parent):
return len(self._header_labels)
def data(self, index, role):
if role == Qt.DisplayRole:
row = index.row()
col = index.column()
return self._data[row][col]
def headerData(self, section, orientation, role):
if role == Qt.DisplayRole:
if orientation == Qt.Horizontal:
return self._header_labels[section]
def sort(self, column, order):
self.layoutAboutToBeChanged.emit()
self._data = sorted(self._data, key=lambda x: x[column], reverse=order == Qt.DescendingOrder)
self.layoutChanged.emit()
在上述代码中，我们创建了一个名为CustomTableModel的自定义表格模型，继承自QAbstractTableModel。CustomTableModel中重写了以下几个方法：
rowCount方法返回表格的行数；columnCount方法返回表格的列数；data方法返回指定单元格的数据；headerData方法返回指定表头的数据；sort方法对表格进行排序。
4.4. 使用自定义表格模型
最后，我们需要将创建的自定义表格模型应用到表格上，以启用自动排序功能。首先，需要将QTableWidget的声明改为QTableView，并将表格模型作为参数传递给QTableView的构造函数。例如，以下代码使用CustomTableModel替换之前的QTableWidget：
from PyQt5.QtWidgets import QTableView
model = CustomTableModel(data, header_labels)
view = QTableView()
view.setModel(model)
至此，我们已经完成了自动排序功能的实现。点击”按姓名排序”按钮后，表格将会按照姓名的字母顺序进行排序。你可以按照类似的方式实现其他属性的排序功能。
5. 运行示例
为了演示自动排序功能，我们可以创建一个名称、年龄和性别都可以编辑的表格，并对姓名属性进行排序。
完整代码如下：
import sys
from PyQt5.QtWidgets import QApplication, QTableWidget, QTableWidgetItem, QVBoxLayout, QWidget, QPushButton, QTableView
from PyQt5.QtCore import Qt, QAbstractTableModel
class CustomTableModel(QAbstractTableModel):
def __init__(self, data, header_labels, parent=None):
super().__init__(parent)
self._data = data
self._header_labels = header_labels
def rowCount(self, parent):
return len(self._data)
def columnCount(self, parent):
return len(self._header_labels)
def data(self, index, role):
if role == Qt.DisplayRole:
row = index.row()
col = index.column()
return self._data[row][col]
def headerData(self, section, orientation, role):
if role == Qt.DisplayRole:
if orientation == Qt.Horizontal:
return self._header_labels[section]
def sort(self, column, order):
self.layoutAboutToBeChanged.emit()
self._data = sorted(self._data, key=lambda x: x[column], reverse=order == Qt.DescendingOrder)
self.layoutChanged.emit()
app = QApplication(sys.argv)
data = [
["小明", "18", "男"],
["小红", "22", "女"],
["小刚", "20", "男"]
]
header_labels = ["姓名", "年龄", "性别"]
model = CustomTableModel(data, header_labels)
view = QTableView()
view.setModel(model)
sort_button = QPushButton("按姓名排序")
widget = QWidget()
layout = QVBoxLayout(widget)
layout.addWidget(sort_button)
layout.addWidget(view)
def sort_by_name():
view.horizontalHeader().sortIndicatorChanged.emit(0, Qt.AscendingOrder)
sort_button.clicked.connect(sort_by_name)
widget.show()
sys.exit(app.exec_())
点击运行后，将会弹出一个GUI窗口，其中包含一个表格和一个排序按钮。点击按钮后，表格将按照姓名进行排序。
这就是使用Python的QTableWidget实现自动排序的方法。通过创建自定义的表格模型并重写排序相关的方法，我们可以轻松实现表格的自动排序功能，提升用户体验。