闭包 (closure) 是函数式编程中的一个概念,Python虽然不是纯粹的是函数式编程语言,但它仍然具有函数式编程的一些特性。在以前的文章Python笔记:Python装饰器中介绍的装饰器其实就使用了闭包,本文来介绍一下Python中的闭包。
目录介绍闭包之前,先理解什么是嵌套函数和变量作用域。
函数嵌套就是在一个函数内部又定义函数:
def TestFunc(val1): val = val1 def innerFunc(): print(val) innerFunc() if __name__ == '__main__': TestFunc('Hello world')
在TestFunc()内部又定义了innerFunc()函数,函数的嵌套主要两个方面的作用。
在函数内部定义的变量称为局部变量,只在函数内部有效。一旦函数执行完毕,局部变量就会被回收。全局变量是定义在函数外的变量,作用域范围为全局,函数内部和外部都可以访问它。
一般情况下,不能在函数内部改变全局变量的值。如果想要在函数内部改变全局变量的值,需要加上 global 关键字:
VALUE1 = 666 def TestFunc2(): global VALUE1 VALUE1 = VALUE1 + 1 print(VALUE1) TestFunc2() print(VALUE1)
执行结果:
667 667
如果函数内部局部变量和全局变量同名,那么在函数内部,局部变量会覆盖全局变量:
VALUE1 = 666 def TestFunc3(): VALUE1 = 2 print(VALUE1) TestFunc3() print(VALUE1)
执行结果:
2 666
类似的,对于嵌套函数,内部函数可以访问外部函数定义的变量,但是无法修改,如果要修改,需要加上 nonlocal 关键字:
def TestFunc(): val = 1 def innerFunc(): nonlocal val val = 2 print("inner:",val) innerFunc() print("outer:", val) TestFunc()
执行结果:
inner: 2 outer: 2
同样,如果内部函数的变量和外部函数变量同名,内部函数变量会覆盖外部函数的变量。
闭包和嵌套函数类似,不同之处在于外部函数返回的是一个函数,而不是一个具体的值。下面用闭包实现计算一个数的 n 次幂:
def nth_power(exponent): def exponent_of(base): return base ** exponent return exponent_of square = nth_power(2) # 平方 cube = nth_power(3) # 立方 print(square) print(cube) print(square(2)) # 2的平方 print(cube(2)) # 2的立方
执行结果:
<function nth_power.<locals>.exponent_of at 0x000001E9629C04C8> <function nth_power.<locals>.exponent_of at 0x000001E9629C0708> 4 8
外部函数 nth_power() 返回的是函数 exponent_of()函数,闭包是一个函数对象,它记住了封闭作用域中的值,也就是记住了外部函数 nth_power() 的参数 exponent。
函数exponent_of()的作用域只在nth_power()内部,通过使用闭包,扩展了它的作用域,使能够在其作用域之外调用内部函数。即使函数本身从当前命名空间中删除,闭包作用域中的这个值也会被记住:
def nth_power(exponent): def exponent_of(base): return base ** exponent return exponent_of square = nth_power(2) # 平方 print(square(2)) # 2的平方 del nth_power print(square(3)) # 3的平方
执行结果:
4 9
闭包函数对象有一个__closure__
属性,返回cell 对象的元组,而cell 对象中保存闭包的变量(例子中的exponent值):
print(square) print(cube) print(square.__closure__) print(square.__closure__[0].cell_contents) print(cube.__closure__[0].cell_contents)
执行结果:
<function nth_power.<locals>.exponent_of at 0x000001DC127D04C8> <function nth_power.<locals>.exponent_of at 0x000001DC127D0708> (<cell at 0x000001DC124401F8: int object at 0x00007FFFA7FAA1B0>,) 2 3
和嵌套函数返回一个值不同,闭包返回的是一个函数对象。闭包和嵌套函数的优点是可以让程序变得更简洁易读。python装饰器(decorator)中也使用闭包,可参考:Python笔记:Python装饰器。
欢迎关注公众号:「测试开发小记」及时接收最新技术文章!