我们之前所看到的函数都是在全局范围内定义的,它们都是全局函数。Python 还支持在函数体内定义函数,这种被放在函数体内定义的函数称为局部函数。
在默认情况下,局部函数对外部是隐藏的,局部函数只能在其封闭(enclosing)函数内有效,其封闭函数也可以返回局部函数,以便程序在其他作用域中使用局部函数。
def get_math_func(type, nn) :
# 定义一个计算平方的局部函数
def square(n) : # ①
return n * n
# 定义一个计算立方的局部函数
def cube(n) : # ②
return n * n * n
# 定义一个计算阶乘的局部函数
def factorial(n) : # ③
result = 1
for index in range(2, n + 1) :
result *= index
return result
# 调用局部函数
if type == "square" :
return square(nn)
elif type == "cube":
return cube(nn)
else:
return factorial(nn)
print(get_math_func("square", 3)) # 输出9
print(get_math_func("cube", 3)) # 输出27
print(get_math_func("", 3)) # 输出6
上面程序中第一行粗体字代码定义了 get_math_func() 函数,接下来程序的 ①、②、③ 号代码定义了 3 个局部函数,而 get_math_func() 函数则根据参数选择调用不同的局部函数。
如果封闭函数没有返回局部函数,那么局部函数只能在封闭函数内部调用,如上面程序所示。另外,还会出现一种情况,如果封闭函数将局部函数返回,且程序使用变量保存了封闭函数的返回值,那么这些局部函数的作用域就会被扩大。因此程序完全可以自由地调用它们,就像它们都是全局函数一样(函数返回函数的情况后续章节会详细介绍)。
局部函数内的变量也会遮蔽它所在函数内的局部变量(这句话有点拗口),请看如下代码:
def foo ():
# 局部变量name
name = 'Charlie'
def bar ():
# 访问bar函数所在的foo函数的name局部变量
print(name) # Charlie
name = '孙悟空'
bar()
foo()
运行上面代码,会导致如下错误:
UnboundLocalError: local variable 'name' referenced before assignment
该错误是由局部变量遮蔽局部变量导致的,在 bar() 函数中定义的 name 局部变量遮蔽了它所在 foo() 函数内的 name 局部变量,因此导致程序报错。 为了声明 bar() 函数中的“name='孙悟空'”赋值语句不是定义新的局部变量,只是访问它所在 foo() 函数内的 name 局部变量,Python 提供了 nonlocal 关键字,通过 nonlocal 语句即可声明访问赋值语句只是访问该函数所在函数内的局部变量。将上面程序改为如下形式: 增加上面程序中第 5 行之后,接下来 bar() 函数中的“name='孙悟空'”就不再是定义新的局部变量,而是访问它所在函数(foo() 函数)内的 name 局部变量。def foo ():
# 局部变量name
name = 'Charlie'
def bar ():
nonlocal name
# 访问bar函数所在的foo函数的name局部变量
print(name) # Charlie
name = '孙悟空'
bar()
foo()