Python中的闭包、nonlocal

要了解Python关键字nonlocal的用法，我们要先搞清楚什么是闭包，以及为什么要使用闭包。

闭包

闭包的定义：能够读取其他函数内部变量的函数。

我们现在有一个需要：要设计一个avg的函数，作用是计算不断增加的值的平均值。

def mark_avg():
    nums = []

    def average(num):
        nums.append(num)
        return sum(nums) / len(nums)

    return average


avg = mark_avg()

print(avg(1))
print(avg(2))
print(avg(3))

# 输出 1.0 1.5 2.0

上例中的mark_avg函数中的函数体部分就是闭包，我们为了在特定的函数外加上一个类函数全局变量nums列表，而使用了闭包。nums变量被称为自由变量（指未在本地作用域中绑定的变量），这是一个比较难理解的概念，nums在我们第一次调用avg = mark_avg()的时候，在mark_avg里已经返回结束，nums的作用域就消失了，可是我们知道，我们每次调用avg(1) avg(2)函数时，明明nums列表里还存着值。这就是闭包的作用，会保留定义函数时存在的自由变量的绑定。

nonloacl（同global）

但是这里有一个问题，就是nums是一个列表，为可变变量，我们只是append往它上面新增数据，并没有改变它的引用，所以它不会出问题。看下面的例子：

def mark_avg():
    count = 0
    total = 0

    def average(num):
        count += 1
        total += num
        return total / count

    return average


avg = mark_avg()

print(avg(1))

# error: UnboundLocalError: local variable 'count' referenced before assignment

上面这个例子我们会发现报错，很容易我们就知道，因为count total是不可变int类型，+=操作会为我们重新引用变量，当我们执行count += 1时，其实在函数average内部创建了一个count局部变量，把原来的自由变量count的引用覆盖了，这时我们就引用不到原来的自由变量count，那这个问题要怎么解决呢，Python给我们引入一个关键字：nonlocal，我们可以把上面的例子修改如下：

def mark_avg():
    count = 0
    total = 0

    def average(num):
        nonlocal count, total
        count += 1
        total += num
        return total / count

    return average


avg = mark_avg()

print(avg(1))
print(avg(2))
print(avg(3))

# 输出：1.0 1.5 2.0

我们在average函数内部定义了nonlocal count, total，这样就可以正常作用自由变量。