Python 如何使用 lambda 函数进行排序

在本文中，我们将介绍如何使用 Python 中的 lambda 函数进行排序。lambda 函数是一种特殊的匿名函数，它可以简化代码并提供一种便捷的方式来定义简单的函数。Python 的内置函数 sorted() 可以接受自定义排序函数作为参数，而 lambda 函数正好可以满足这个要求。

阅读更多：Python 教程

什么是 lambda 函数

lambda 函数是一种匿名函数，它可以在一行代码中定义一个简单的函数。它的语法如下：

lambda arguments: expression

lambda 函数可以有多个参数，用逗号分隔。expression 是一个只有一行的表达式，并且是函数的返回值。

下面是一个简单的 lambda 函数的例子，它接受两个参数并返回它们的和：

sum = lambda a, b: a + b
print(sum(5, 3))  # 输出：8

在这个例子中，lambda a, b: a + b 定义了一个 lambda 函数，它接受两个参数 a 和 b，并返回它们的和。通过调用 sum(5, 3)，我们可以得到这两个参数的和，即 8。

利用 lambda 函数进行排序

Python 的内置函数 sorted() 可以用来对列表进行排序。它可以接受一个可迭代对象作为输入，并返回一个新的已排序的列表。sorted() 函数还可以接受一个可选的参数 key，用于指定自定义的排序函数。而 lambda 函数则可以方便地定义这样的排序函数。

下面是一个例子，使用 lambda 函数对一个列表进行排序：

fruits = ['apple', 'banana', 'cherry', 'durian']
sorted_fruits = sorted(fruits, key=lambda x: x[0])
print(sorted_fruits)  # 输出：['apple', 'banana', 'cherry', 'durian']

在这个例子中，sorted() 函数使用了一个 lambda 函数作为 key 参数。lambda 函数 lambda x: x[0] 定义了一个规则，按照每个元素的第一个字符进行排序。通过调用 sorted(fruits, key=lambda x: x[0])，我们可以得到一个按照首字母排序的新列表 ['apple', 'banana', 'cherry', 'durian']。

lambda 函数也可以被用于对复杂对象的某个属性进行排序。例如，我们可以用 lambda 函数对一个包含学生信息的列表进行排序，按照学生的分数进行升序排序：

students = [{'name': 'Alice', 'score': 90},
            {'name': 'Bob', 'score': 80},
            {'name': 'Charlie', 'score': 95}]
sorted_students = sorted(students, key=lambda x: x['score'])
print(sorted_students)
# 输出：[{'name': 'Bob', 'score': 80}, {'name': 'Alice', 'score': 90}, {'name': 'Charlie', 'score': 95}]

在这个例子中，lambda 函数 lambda x: x['score'] 定义了一个规则，按照学生字典中的 score 属性进行排序。通过调用 sorted(students, key=lambda x: x['score'])，我们可以得到一个按照学生分数升序排序的新列表。

特殊排序需求

使用 lambda 函数进行排序时，我们还可以根据特定的需求进行更加灵活的排序。

降序排序

默认情况下，sorted() 函数进行的是升序排序。如果我们想要进行降序排序，可以利用 lambda 函数的返回值。

numbers = [3, 1, 4, 1, 5, 9, 2, 6, 5]
sorted_numbers = sorted(numbers, key=lambda x: -x)
print(sorted_numbers)  # 输出：[9, 6, 5, 5, 4, 3, 2, 1, 1]

在这个例子中，lambda 函数 lambda x: -x 定义了一个规则，通过返回值取负数，即将默认的升序排序转为降序排序。

多个条件排序

使用 lambda 函数进行多个条件排序时，我们可以在 lambda 函数中使用多个表达式来定义排序规则。

students = [{'name': 'Alice', 'score': 90, 'age': 18},
            {'name': 'Bob', 'score': 80, 'age': 20},
            {'name': 'Charlie', 'score': 95, 'age': 19}]
sorted_students = sorted(students, key=lambda x: (x['score'], -x['age']))
print(sorted_students)
# 输出：[{'name': 'Bob', 'score': 80, 'age': 20}, {'name': 'Charlie', 'score': 95, 'age': 19}, {'name': 'Alice', 'score': 90, 'age': 18}]

在这个例子中，lambda 函数 lambda x: (x['score'], -x['age']) 定义了一个规则，先按照学生的分数升序排序，如果分数相同，则按照年龄降序排序。通过调用 sorted(students, key=lambda x: (x['score'], -x['age']))，我们可以得到一个按照学生分数升序、年龄降序排序的新列表。

总结

使用 lambda 函数进行排序是一种简单、方便的方法。通过在 sorted() 函数的 key 参数中使用 lambda 函数，我们可以灵活地定义排序规则。除了基本的升序排序，lambda 函数还可以实现降序排序和多个条件排序。了解和掌握 lambda 函数的用法将有助于我们更好地利用 Python 进行排序操作。