Python的函數(shù)是“一等公民”,因此函數(shù)本身也是一個(gè)對(duì)象,函數(shù)既可用于賦值,也可用作其他函數(shù)的參數(shù),還可作為其他函數(shù)的返回值。
使用函數(shù)變量
python的函數(shù)也是一種值:所有函數(shù)都是function對(duì)象,這意味著可以把函數(shù)本身賦值給變量,就像把整數(shù)、浮點(diǎn)數(shù)、列表、元組賦值給變量一樣。
當(dāng)把函數(shù)賦值給變量之后,接下來程序也可通過該變量來調(diào)用函數(shù)。例如如下代碼:
#定義一個(gè)計(jì)算乘方的函數(shù)
defpow(base,exponent):
result=1
foriinrange(1,exponent+1):
result*=base
returnresult
#將pow函數(shù)賦值給my_fun,則my_fun可當(dāng)成pow使用
my_fun=pow
print(my_fun(3,4))#輸出81
#定義一個(gè)計(jì)算面積的函數(shù)
defarea(width,height):
returnwidth*height
#將area函數(shù)賦值給my_fun,則my_fun可當(dāng)成area使用
my_fun=area
print(my_fun(3,4))#輸出12
從上面代碼可以看出,程序依次將pow()、area()函數(shù)賦值給my_fun變量,接下來即可通過my_fun變量分別調(diào)用pow()、area()函數(shù)。
其實(shí)python已經(jīng)內(nèi)置了計(jì)算乘方的方法,因此此處的pow()函數(shù)并沒有太大的實(shí)際意義,只是作為示范使用。
通過對(duì)my_fun變量賦值不同的函數(shù),可以讓my_fun在不同的時(shí)間指向不同的函數(shù),從而讓程序更加靈活。由此可見,使用函數(shù)變量的好處是讓程序更加靈活。
除此之外,程序還可使用函數(shù)作為另一個(gè)函數(shù)的形參和(或)返回值。
使用函數(shù)作為函數(shù)形參
有時(shí)候需要定義一個(gè)函數(shù),該函數(shù)的大部分計(jì)算邏輯都能確定,但某些處理邏輯暫時(shí)無法確定,這意昧著某些程序代碼需要?jiǎng)討B(tài)改變,如果希望調(diào)用函數(shù)時(shí)能動(dòng)態(tài)傳入這些代碼,那么就需要在函數(shù)中定義函數(shù)形參,這樣即可在調(diào)用該函數(shù)時(shí)傳入不同的函數(shù)作為參數(shù),從而動(dòng)態(tài)改變這段代碼。
Python支持像使用其他參數(shù)一樣使用函數(shù)參數(shù),例如如下程序:
#定義函數(shù)類型的形參,其中fn是一個(gè)函數(shù)
defmap(data,fn):
result=[]
#遍歷data列表中每個(gè)元素,并用fn函數(shù)對(duì)每個(gè)元素進(jìn)行計(jì)算
#然后將計(jì)算結(jié)果作為新數(shù)組的元素
foreindata:
result.append(fn(e))
returnresult
#定義一個(gè)計(jì)算平方的函數(shù)
defsquare(n):
returnn*n
#定義一個(gè)計(jì)算立方的函數(shù)
defcube(n):
returnn*n*n
#定義一個(gè)計(jì)算階乘的函數(shù)
deffactorial(n):
result=1
forindexinrange(2,n+1):
result*=index
returnresult
data=[3,4,9,5,8]
print("原數(shù)據(jù):",data)
#下面程序代碼3次調(diào)用map()函數(shù),每次調(diào)用時(shí)傳入不同的函數(shù)
print("計(jì)算數(shù)組元素的平方")
print(map(data,square))
print("計(jì)算數(shù)組元素的立方")
print(map(data,cube))
print("計(jì)算數(shù)組元素的階乘")
print(map(data,factorial))
上面程序中定義了一個(gè)map()函數(shù),該函數(shù)的第二個(gè)參數(shù)是一個(gè)函數(shù)類型的參數(shù),這意味著每次調(diào)用函數(shù)時(shí)可以動(dòng)態(tài)傳入一個(gè)函數(shù),隨著實(shí)際傳入函數(shù)的改變,就可以動(dòng)態(tài)改變map()函數(shù)中的部分計(jì)算代碼。
Pthon3本身也提供了一個(gè)map()函數(shù),Python3內(nèi)置的map()函數(shù)的功能和此處定義的map()函數(shù)功能類似,但更加強(qiáng)大。
接下來的三行粗體字代碼調(diào)用了map()函數(shù)三次,三次調(diào)用依次傳入了square、cube、factorial函數(shù)作為參數(shù),這樣每次調(diào)用map()函數(shù)時(shí)實(shí)際的執(zhí)行代碼是有區(qū)別的。
編譯、運(yùn)行上面程序,可以看到如下輸出結(jié)果:
原數(shù)據(jù):[3,4,9,5,8]
計(jì)算數(shù)組元素的平方
[9,16,81,25,64]
計(jì)算數(shù)組元素的立方
[27,64,729,125,512]
計(jì)算數(shù)組元素的階乘
[6,24,362880,120,40320]
從上面介紹不難看出,通過使用函數(shù)作為參數(shù)可以在調(diào)用函數(shù)時(shí)動(dòng)態(tài)傳入函數(shù),實(shí)際上就可以動(dòng)態(tài)改變被調(diào)用函數(shù)的部分代碼。
在程序最后添加如下一行:
#獲取map的類型
print(type(map))
運(yùn)行上面代碼,將會(huì)看到如下輸出結(jié)果:
《class'function'>
使用函數(shù)作為返回值
前面己經(jīng)提到,Python還支持使用函數(shù)作為其他函數(shù)的返回值。例如如下程序:
defget_math_func(type):
#定義一個(gè)計(jì)算平方的局部函數(shù)
defsquare(n):#①
returnn*n
#定義一個(gè)計(jì)算立方的局部函數(shù)
defcube(n):#②
returnn*n*n
#定義一個(gè)計(jì)算階乘的局部函數(shù)
deffactorial(n):#③
result=1
forindexinrange(2,n+1):
result*=index
returnresult
#返回局部函數(shù)
iftype=="square":
returnsquare
iftype=="cube":
returncube
else:
returnfactorial
#調(diào)用get_math_func(),程序返回一個(gè)嵌套函數(shù)
math_func=get_math_func("cube")#得到cube函數(shù)
print(math_func(5))#輸出125
math_func=get_math_func("square")#得到square函數(shù)
print(math_func(5))#輸出25
math_func=get_math_func("other")#得到factorial函數(shù)
print(math_func(5))#輸出120
程序中,定義了一個(gè)get_math_func()函數(shù),該函數(shù)將返回另一個(gè)函數(shù)。接下來在get_math_func()函數(shù)體內(nèi)的①、②、③號(hào)粗體字代碼分別定義了三個(gè)局部函數(shù),最后get_math_func()函數(shù)會(huì)根據(jù)所傳入的參數(shù),使用這三個(gè)局部函數(shù)之一作為返回值。
在定義了會(huì)返回函數(shù)的get_math_func()函數(shù)之后,接下來程序調(diào)用get_math_func()函數(shù)時(shí)即可返回所需的函數(shù),如上面程序中代碼所示。
以上內(nèi)容為大家介紹了python培訓(xùn)之函數(shù)的高級(jí)使用方法,希望對(duì)大家有所幫助,如果想要了解更多Python相關(guān)知識(shí),請(qǐng)關(guān)注IT培訓(xùn)機(jī)構(gòu):千鋒教育。