函数式编程 之

返回函数

• 函数可以返回具体的值
• 也可以返回一个函数作为结果

1. 引子

1.1 定义一个普通函数

>>> def func():

...   print("abc")

...   return None

...

>>> my_func = func()

abc

>>> print(my_func)

None

>>> my_func

>>>

1.2 函数作为返回值返回

• 被返回的函数在函数体内定义

In [1]: def out_func():

   ...:     def in_func():

   ...:         print("in func")

   ...:         return 100

   ...:     return in_func

   ...:

   ...:

In [2]: f = out_func()

In [3]: print(type(f))

<class 'function'>

In [4]: print(f)

<function out_func.<locals>.in_func at 0x000001B9B99D6BF8>

In [5]: f()

in func

Out[5]: 100

In [6]:

1.3 带参数列表的返回函数

>>> def out_func(*args):

...   def in_func():

...     rst = 0

...     for i in args:

...       rst += i

...     return rst

...   return in_func

...

>>> f1 = out_func(1, 2, 3, 4, 5)

>>> f1()

15

>>> f2 = out_func(6, 7, 8, 9, 10)

>>> f2()

40

2. 闭包 closure

• 当一个函数在内部定义函数，并且内部函数应用外部函数的参数或者局部变量时，内部函数被当做返回值之时，相关参数和变量会保存在返回的函数中，这种结果，叫闭包
• 上方 1.3 带参数列表的返回函数 的例子就是一个标准闭包结构

2.1 常见的关于闭包的坑

>>> def out_func():

...   res = []

...   for i in range(1, 4):

...     def in_func():

...       return i * i

...     res.append(in_func)

...   return res

...

>>> f1, f2, f3 = out_func()

>>> f1()

9

>>> f2()

9

>>> f3()

9

>>>

2.2 造成上述状况的原因

• 返回函数引用了变量 i，i 并非立即执行，而是等到三个函数都返回的时候才统一使用，此时 i 已经变成了 3，最终调用的时候，返回的都是 3*3

• 由此可见：返回闭包时，返回函数不能引用任何循环变量

2.3 解决方案

• 再创建一个函数，用该函数的参数绑定循环变量的当前值，无论该循环变量以后如何改变，已经绑定的函数参数值不再改变

>>> def out_func():

...   def in1(n):

...     def in2():

...       return n * n

...     return in2

...   res = []

...   for i in range(1, 4):

...     res.append(in1(i))

...   return res

...

>>> f1, f2, f3 = out_func()

>>> f1()

1

>>> f2()

4

>>> f3()

9

>>>