函数作为返回值
高阶函数除了可以接受函数作为参数外,还可以把函数作为结果值返回。
我们来实现一个可变参数的求和。通常情况下,求和的函数是这样定义的:
>>> def sum(*args):
... s=0
... for x in args:
... s=s+x
... return s
...
但是,如果不需要立刻求和,而是在后面的代码中,根据需要再计算怎么办?可以不返回求和的结果,而是返回求和的函数:
>>> def lazysum(*args):
... def sum():
... r=0
... for x in args:
... r=r+x
... return r
... return sum
...
当我们调用lazysum()
时,返回的并不是求和结果,而是求和函数 调用求和函数f1的时候才返回值
>>> f1=lazysum(1,2,3)
>>> f1
<function lazysum.<locals>.sum at 0x0000000002916048>
>>> f1()
6
>>> f2=lazysum(1,2,3)
>>> f2
<function lazysum.<locals>.sum at 0x0000000002907EA0>
>>> f2()
6
并且虽然参数和方法一致。但是返回的都会是新的函数,函数地址不一致
在这个例子中,我们在函数lazysum
中又定义了函数sum
,并且,内部函数sum
可以引用外部函数lazysum
的参数和局部变量,当lazysum
返回函数sum
时,相关参数和变量都保存在返回的函数中,这种称为“闭包(Closure)”
返回的函数并没有立刻执行,而是直到调用了该函数f()
才执行
>>> def count():
... fs=[]
... for i in range(1,4):
... print(i)
... def f():
... return i*i
... fs.append(f)
... return fs
...
>>> f1,f2,f3=count()
1
2
3
>>> f1()
9
>>> f2()
9
>>> f3()
9
如上 返回的函数f()引用了变量i
,但它并非立刻执行 在执行具体f()函数的时候 i已经循环完毕都变成了3
返回闭包时牢记一点:返回函数不要引用任何循环变量,或者后续会发生变化的变量。
lambda
当我们在传入函数时,有些时候,不需要显式地定义函数,直接传入匿名函数更方便。
在Python中,对匿名函数提供了有限支持。还是以map()
函数为例,计算f(x)=x2时,除了定义一个f(x)
的函数外,还可以直接传入匿名函数:
>>> list(map(lambda x: x * x, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
[1, 4, 9, 16, 25, 36, 49, 64, 81]
通过对比可以看出,匿名函数lambda x: x * x
实际上就是:
def f(x):
return x * x
关键字lambda
表示匿名函数,冒号前面的x
表示函数参数。
匿名函数有个限制,就是只能有一个表达式,不用写return
,返回值就是该表达式的结果。
用匿名函数有个好处,因为函数没有名字,不必担心函数名冲突。此外,匿名函数也是一个函数对象,也可以把匿名函数赋值给一个变量,再利用变量来调用该函数:
>>> f = lambda x: x * x
>>> f
<function <lambda> at 0x101c6ef28>
>>> f(5)
25
同样,也可以把匿名函数作为返回值返回,比如:
def build(x, y):
return lambda: x * x + y * y