函数式编程 - 匿名函数

优质

小牛编辑

140浏览

2023-12-01

在 Python 中，我们使用 def 语句来定义函数，比如：

def double(x):
    return 2 * x

除了用上面的方式定义函数，Python 还提供了一个关键字 lambda，让我们可以创建一个匿名函数，也就是没有名称的函数。它的形式如下：

lambda 参数: 表达式

关键字 lambda 说明它是一个匿名函数，冒号 : 前面的变量是该匿名函数的参数，冒号后面是函数的返回值，注意这里不需使用 return 关键字。

我们将上面的 double 函数改写成一个匿名函数，如下：

lambda x: 2 * x

那怎么调用匿名函数呢？可以直接这样使用：

>>> (lambda x: 2 * x)(8)
16

由于匿名函数本质上是一个函数对象，也可以将其赋值给另一个变量，再由该变量来调用函数，如下：

>>> f = lambda x: 2 * x   # 将匿名函数赋给变量 f  
>>> f
<function <lambda> at 0x7f835a696578>
>>> f(8)
16

使用场景

lambda 函数一般适用于创建一些临时性的，小巧的函数。比如上面的 double 函数，我们当然可以使用 def 来定义，但使用 lambda 来创建会显得很简洁，尤其是在高阶函数的使用中。

看一个例子：

def func(g, arr):
    return [g(x) for x in arr]

现在给一个列表 [1, 2, 3, 4]，利用上面的函数，对列表中的元素加 1，返回一个新的列表，你可能这样用：

def add_one(x):
    return x + 1
arr = func(add_one, [1, 2, 3, 4])

这样做没什么错，可是 add_one 这个函数太简单了，使用 def 定义未免有点小题大作，我们改用 lambda：

arr = func(lambda x: x + 1, [1, 2, 3, 4])

是不是很简洁、易懂？