Python | lambda匿名函数

最近看到一些进阶教程，发现都很自然而然地应用了lambda函数，这里总结一下lambda函数的定义和用法。

lambda函数定义

Python 使用 lambda 来创建匿名函数。

lambda 函数是一种小型、匿名的、内联函数，它可以具有任意数量的参数，但只能有一个表达式。

匿名函数不需要使用 def 关键字定义完整函数。

lambda 函数通常用于编写简单的、单行的函数，通常在需要函数作为参数传递的情况下使用，例如在 map()、filter()、reduce() 等函数中。

即：如果一个函数的函数体仅有 1 行表达式，则该函数就可以用 lambda 表达式来代替。

lambda 函数特点

• lambda 函数是匿名的，它们没有函数名称，只能通过赋值给变量或作为参数传递给其他函数来使用。
• lambda 函数通常只包含一行代码，这使得它们适用于编写简单的函数。

lambda 语法格式

lambda arguments: expression

• lambda是 Python 的关键字，用于定义 lambda 函数。
• arguments 是参数列表，可以包含零个或多个参数，但必须在冒号(:)前指定。
• expression 是一个表达式，用于计算并返回函数的结果。

举例

没有参数

以下的 lambda 函数没有参数：

f = lambda: "Hello, world!" 
print(f())  # 输出: Hello, world!

输出结果为：

Hello, world!

一个参数

以下实例使用 lambda 创建匿名函数，设置一个函数参数 a，函数计算参数 a 加 10，并返回结果：

x = lambda a : a + 10 
print(x(5))  # 输出：15

多个参数

lambda 函数也可以设置多个参数，参数使用逗号 , 隔开：

以下实例使用 lambda 创建匿名函数，函数参数 a 与 b 相乘，并返回结果：

x = lambda a, b : a * b 
print(x(5, 6))  # 输出：30

以下实例使用 lambda 创建匿名函数，函数参数 a、b 与 c 相加，并返回结果：

x = lambda a, b, c : a + b + c 
print(x(5, 6, 2))  # 输出：13

与内置函数一起使用

lambda 函数通常与内置函数如 map()、filter() 和 reduce() 一起使用，以便在集合上执行操作。

map()函数

map() 会根据提供的函数对指定序列做映射。

第一个参数 function 以参数序列中的每一个元素调用 function 函数，返回包含每次 function 函数返回值的新列表（python 2.x是列表，3.x是map迭代器）。

map() 函数语法：

map(function, iterable, ...)

• function – 函数
• iterable – 一个或多个序列

>>> def square(x) :         # 计算平方数
...     return x ** 2
... 
>>> map(square, [1,2,3,4,5])    # 计算列表各个元素的平方
<map object at 0x100d3d550>     # 返回迭代器
>>> list(map(square, [1,2,3,4,5]))   # 使用 list() 转换为列表
[1, 4, 9, 16, 25]
>>> list(map(lambda x: x ** 2, [1, 2, 3, 4, 5]))   # 使用 lambda 匿名函数
[1, 4, 9, 16, 25]

filter()函数

filter() 函数用于过滤序列，过滤掉不符合条件的元素，返回由符合条件元素组成的新列表。

该接收两个参数，第一个为函数，第二个为序列，序列的每个元素作为参数传递给函数进行判断，然后返回 True 或 False，最后将返回 True 的元素放到新列表中。

filter() 方法的语法：

filter(function, iterable)

• function – 判断函数。
• iterable – 可迭代对象。

numbers = [1, 2, 3, 4, 5, 6, 7, 8]
even_numbers = list(filter(lambda x: x % 2 == 0, numbers))
print(even_numbers)  # 输出：[2, 4, 6, 8]

reduce()函数

reduce() 函数会对参数序列中元素进行累积。

函数将一个数据集合（链表，元组等）中的所有数据进行下列操作：用传给 reduce 中的函数 function（有两个参数）先对集合中的第 1、2 个元素进行操作，得到的结果再与第三个数据用 function 函数运算，最后得到一个结果。

**注意：**Python3.x reduce() 已经被移到 functools 模块里，如果我们要使用，需要引入 functools 模块来调用 reduce() 函数：

from functools import reduce

reduce() 函数语法：

reduce(function, iterable[, initializer])

• function – 函数，有两个参数
• iterable – 可迭代对象
• initializer – 可选，初始参数

from functools import reduce

def add(x, y) :            # 两数相加
    return x + y
sum1 = reduce(add, [1,2,3,4,5])   # 计算列表和：1+2+3+4+5
sum2 = reduce(lambda x, y: x+y, [1,2,3,4,5])  # 使用 lambda 匿名函数
product = reduce(lambda x, y: x*y, [1,2,3,4,5] )  # 使用lambda 匿名函数计算列表乘积
print(sum1)  # 输出：15
print(sum2)  # 输出：15
print(product)  # 输出：120

参考资料

1. Python map() 函数
2. Python lambda（匿名函数）
3. Python filter() 函数
4. Python reduce() 函数