1. 程式人生 > >Python(九)語法 函數語言程式設計(函式作為返回值,lambda)

Python(九)語法 函數語言程式設計(函式作為返回值,lambda)

函式作為返回值

高階函式除了可以接受函式作為引數外,還可以把函式作為結果值返回。

我們來實現一個可變引數的求和。通常情況下,求和的函式是這樣定義的:

>>> def sum(*args):
...  s=0
...  for x in args:
...   s=s+x
...  return s
...

但是,如果不需要立刻求和,而是在後面的程式碼中,根據需要再計算怎麼辦?可以不返回求和的結果,而是返回求和的函式:

>>> def lazysum(*args):
...  def sum():
...   r=0
...   for x in args:
...    r=r+x
...   return r
...  return sum
...

當我們呼叫lazysum()時,返回的並不是求和結果,而是求和函式 呼叫求和函式f1的時候才返回值

>>> f1=lazysum(1,2,3)
>>> f1
<function lazysum.<locals>.sum at 0x0000000002916048>
>>> f1()
6
>>> f2=lazysum(1,2,3)
>>> f2
<function lazysum.<locals>.sum at 0x0000000002907EA0>
>>> f2()
6

並且雖然引數和方法一致。但是返回的都會是新的函式,函式地址不一致

在這個例子中,我們在函式lazysum中又定義了函式sum,並且,內部函式sum可以引用外部函式lazysum的引數和區域性變數,當lazysum返回函式sum時,相關引數和變數都儲存在返回的函式中,這種稱為“閉包(Closure)”

返回的函式並沒有立刻執行,而是直到呼叫了該函式f()才執行

>>> def count():
...  fs=[]
...  for i in range(1,4):
...   print(i)
...   def f():
...    return i*i
...   fs.append(f)
...  return fs
...
>>> f1,f2,f3=count()
1
2
3
>>> f1()
9
>>> f2()
9
>>> f3()
9

如上 返回的函式f()引用了變數i,但它並非立刻執行 在執行具體f()函式的時候 i已經迴圈完畢都變成了3

返回閉包時牢記一點:返回函式不要引用任何迴圈變數,或者後續會發生變化的變數。

lambda

當我們在傳入函式時,有些時候,不需要顯式地定義函式,直接傳入匿名函式更方便。

在Python中,對匿名函式提供了有限支援。還是以map()函式為例,計算f(x)=x2時,除了定義一個f(x)的函式外,還可以直接傳入匿名函式:

>>> list(map(lambda x: x * x, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
[1, 4, 9, 16, 25, 36, 49, 64, 81]

通過對比可以看出,匿名函式lambda x: x * x實際上就是:

def f(x):
    return x * x

關鍵字lambda表示匿名函式,冒號前面的x表示函式引數。

匿名函式有個限制,就是隻能有一個表示式,不用寫return,返回值就是該表示式的結果。

用匿名函式有個好處,因為函式沒有名字,不必擔心函式名衝突。此外,匿名函式也是一個函式物件,也可以把匿名函式賦值給一個變數,再利用變數來呼叫該函式:

>>> f = lambda x: x * x
>>> f
<function <lambda> at 0x101c6ef28>
>>> f(5)
25

同樣,也可以把匿名函式作為返回值返回,比如:

def build(x, y):
    return lambda: x * x + y * y