Python 自定义排序|极客教程

Python 自定义排序

在 Python 中，我们经常需要对列表、元组等数据类型进行排序操作。Python 的内置函数 sorted() 可以用来对可迭代对象进行排序，但有时候我们需要根据自定义的规则对数据进行排序。本文将介绍如何在 Python 中自定义排序。

自定义排序函数

在 Python 中，我们可以通过自定义排序函数来指定排序的规则。自定义排序函数接受一个元素作为参数，并返回一个用于排序的关键值。Python 中的 sorted() 函数可以接受一个 key 参数，用来指定排序规则。

以下是一个简单的示例，假设我们有一个包含列表的列表，每个列表的第一个元素是学生的姓名，第二个元素是学生的年龄，我们希望按照年龄对学生进行排序：

students = [['Alice', 23], ['Bob', 21], ['Charlie', 25], ['David', 20]]

def sort_by_age(student):
    return student[1]

sorted_students = sorted(students, key=sort_by_age)
print(sorted_students)

输出为：

[['David', 20], ['Bob', 21], ['Alice', 23], ['Charlie', 25]]

降序排序

如果我们希望按照降序排序，可以使用 reverse=True 参数：

sorted_students_desc = sorted(students, key=sort_by_age, reverse=True)
print(sorted_students_desc)

输出为：

[['Charlie', 25], ['Alice', 23], ['Bob', 21], ['David', 20]]

多重排序

有时候我们需要按照多个条件进行排序。可以先根据第一个条件排序，然后再根据第二个条件排序。例如，我们希望按照年龄降序排序，如果年龄相同则按照姓名升序排序：

def sort_by_age_name(student):
    return (-student[1], student[0])

sorted_students_multiple = sorted(students, key=sort_by_age_name)
print(sorted_students_multiple)

输出为：

[['Charlie', 25], ['Alice', 23], ['Bob', 21], ['David', 20]]

使用 lambda 函数

有时候我们可以使用匿名函数 lambda 来定义排序规则，这样可以减少代码的数量。例如，上面的示例可以使用 lambda 函数来实现：

sorted_students_lambda = sorted(students, key=lambda student: (-student[1], student[0]))
print(sorted_students_lambda)

输出为：

[['Charlie', 25], ['Alice', 23], ['Bob', 21], ['David', 20]]

自定义类的排序

除了对基本数据类型进行排序外，我们还可以对自定义类进行排序。如果我们希望按照自定义类的某个属性来排序，可以在自定义类中实现 __lt__() 方法。例如，假设我们有一个 Student 类表示学生，其中包含姓名和年龄两个属性，我们希望按照年龄排序：

class Student:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def __lt__(self, other):
        return self.age < other.age

students_obj = [Student('Alice', 23), Student('Bob', 21), Student('Charlie', 25), Student('David', 20)]
sorted_students_obj = sorted(students_obj)
for student in sorted_students_obj:
    print(student.name, student.age)

输出为：