Python(九)語法 函數語言程式設計(函式作為返回值,lambda)
阿新 • • 發佈:2018-11-20
函式作為返回值
高階函式除了可以接受函式作為引數外,還可以把函式作為結果值返回。
我們來實現一個可變引數的求和。通常情況下,求和的函式是這樣定義的:
>>> 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