Python局部函数及用法

我们之前所看到的函数都是在全局范围内定义的，它们都是全局函数。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 语句即可声明访问赋值语句只是访问该函数所在函数内的局部变量。将上面程序改为如下形式：

def foo ():
    # 局部变量name
    name = 'Charlie'
    def bar ():
     nonlocal name
        # 访问bar函数所在的foo函数的name局部变量
        print(name) # Charlie
        name = '孙悟空'
    bar()
foo()

增加上面程序中第 5 行之后，接下来 bar() 函数中的“name='孙悟空'”就不再是定义新的局部变量，而是访问它所在函数（foo() 函数）内的 name 局部变量。