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