理解Python中函數的參數_python

來源:互聯網
上載者:User

 定義函數的時候,我們把參數的名字和位置確定下來,函數的介面定義就完成了。對於函數的調用者來說,只需要知道如何傳遞正確的參數,以及函數將返回什麼樣的值就夠了,函數內部的複雜邏輯被封裝起來,調用者無需瞭解。

Python的函數定義非常簡單,但靈活度卻非常大。除了正常定義的必選參數外,還可以使用預設參數、可變參數和關鍵字參數,使得函數定義出來的介面,不但能處理複雜的參數,還可以簡化調用者的代碼。
預設參數

我們仍以具體的例子來說明如何定義函數的預設參數。先寫一個計算x2的函數:

def power(x):  return x * x

當我們調用power函數時,必須傳入有且僅有的一個參數x:

>>> power(5)25>>> power(15)225

現在,如果我們要計算x3怎麼辦?可以再定義一個power3函數,但是如果要計算x4、x5……怎麼辦?我們不可能定義無限多個函數。

你也許想到了,可以把power(x)修改為power(x, n),用來計算xn,說幹就幹:

def power(x, n):  s = 1  while n > 0:    n = n - 1    s = s * x  return s

對於這個修改後的power函數,可以計算任意n次方:

>>> power(5, 2)25>>> power(5, 3)125

但是,舊的調用代碼失敗了,原因是我們增加了一個參數,導致舊的代碼無法正常調用:

>>> power(5)Traceback (most recent call last): File "<stdin>", line 1, in <module>TypeError: power() takes exactly 2 arguments (1 given)

這個時候,預設參數就排上用場了。由於我們經常計算x2,所以,完全可以把第二個參數n的預設值設定為2:

def power(x, n=2):  s = 1  while n > 0:    n = n - 1    s = s * x  return s

這樣,當我們調用power(5)時,相當於調用power(5, 2):

>>> power(5)25>>> power(5, 2)25

而對於n > 2的其他情況,就必須明確地傳入n,比如power(5, 3)。

從上面的例子可以看出,預設參數可以簡化函數的調用。設定預設參數時,有幾點要注意:

一是必選參數在前,預設參數在後,否則Python的解譯器會報錯(思考一下為什麼預設參數不能放在必選參數前面);

二是如何設定預設參數。

當函數有多個參數時,把變化大的參數放前面,變化小的參數放後面。變化小的參數就可以作為預設參數。

使用預設參數有什麼好處?最大的好處是能降低調用函數的難度。

舉個例子,我們寫個一年級小學生註冊的函數,需要傳入name和gender兩個參數:

def enroll(name, gender):  print 'name:', name  print 'gender:', gender

這樣,調用enroll()函數只需要傳入兩個參數:

>>> enroll('Sarah', 'F')name: Sarahgender: F

如果要繼續傳入年齡、城市等資訊怎麼辦?這樣會使得調用函數的複雜度大大增加。

我們可以把年齡和城市設為預設參數:

def enroll(name, gender, age=6, city='Beijing'):  print 'name:', name  print 'gender:', gender  print 'age:', age  print 'city:', city

這樣,大多數學生註冊時不需要提供年齡和城市,只提供必須的兩個參數:

>>> enroll('Sarah', 'F')Student:name: Sarahgender: Fage: 6city: Beijing

只有與預設參數不符的學生才需要提供額外的資訊:

enroll('Bob', 'M', 7)enroll('Adam', 'M', city='Tianjin')

可見,預設參數降低了函數調用的難度,而一旦需要更複雜的調用時,又可以傳遞更多的參數來實現。無論是簡單調用還是複雜調用,函數只需要定義一個。

有多個預設參數時,調用的時候,既可以按順序提供預設參數,比如調用enroll('Bob', 'M', 7),意思是,除了name,gender這兩個參數外,最後1個參數應用在參數age上,city參數由於沒有提供,仍然使用預設值。

也可以不按順序提供部分預設參數。當不按順序提供部分預設參數時,需要把參數名寫上。比如調用enroll('Adam', 'M', city='Tianjin'),意思是,city參數用傳進去的值,其他預設參數繼續使用預設值。

預設參數很有用,但使用不當,也會掉坑裡。預設參數有個最大的坑,示範如下:

先定義一個函數,傳入一個list,添加一個END再返回:

def add_end(L=[]):  L.append('END')  return L

當你正常調用時,結果似乎不錯:

>>> add_end([1, 2, 3])[1, 2, 3, 'END']>>> add_end(['x', 'y', 'z'])['x', 'y', 'z', 'END']

當你使用預設參數調用時,一開始結果也是對的:

>>> add_end()['END']

但是,再次調用add_end()時,結果就不對了:

>>> add_end()['END', 'END']>>> add_end()['END', 'END', 'END']

很多初學者很疑惑,預設參數是[],但是函數似乎每次都“記住了”上次添加了'END'後的list。

原因解釋如下:

Python函數在定義的時候,預設參數L的值就被計算出來了,即[],因為預設參數L也是一個變數,它指向對象[],每次調用該函數,如果改變了L的內容,則下次調用時,預設參數的內容就變了,不再是函數定義時的[]了。

所以,定義預設參數要牢記一點:預設參數必須指向不變對象!

要修改上面的例子,我們可以用None這個不變對象來實現:

def add_end(L=None):  if L is None:    L = []  L.append('END')  return L

現在,無論調用多少次,都不會有問題:

>>> add_end()['END']>>> add_end()['END']

為什麼要設計str、None這樣的不變對象呢?因為不變對象一旦建立,對象內部的資料就不能修改,這樣就減少了由於修改資料導致的錯誤。此外,由於對象不變,多任務環境下同時讀取對象不需要加鎖,同時讀一點問題都沒有。我們在編寫程式時,如果可以設計一個不變對象,那就盡量設計成不變對象。
可變參數

在Python函數中,還可以定義可變參數。顧名思義,可變參數就是傳入的參數個數是可變的,可以是1個、2個到任意個,還可以是0個。

我們以數學題為例子,給定一組數字a,b,c……,請計算a2 + b2 + c2 + ……。

要定義出這個函數,我們必須確定輸入的參數。由於參數個數不確定,我們首先想到可以把a,b,c……作為一個list或tuple傳進來,這樣,函數可以定義如下:

def calc(numbers):  sum = 0  for n in numbers:    sum = sum + n * n  return sum

但是調用的時候,需要先組裝出一個list或tuple:

>>> calc([1, 2, 3])14>>> calc((1, 3, 5, 7))84

如果利用可變參數,調用函數的方式可以簡化成這樣:

>>> calc(1, 2, 3)14>>> calc(1, 3, 5, 7)84

所以,我們把函數的參數改為可變參數:

def calc(*numbers):  sum = 0  for n in numbers:    sum = sum + n * n  return sum

定義可變參數和定義list或tuple參數相比,僅僅在參數前面加了一個*號。在函數內部,參數numbers接收到的是一個tuple,因此,函數代碼完全不變。但是,調用該函數時,可以傳入任意個參數,包括0個參數:

>>> calc(1, 2)5>>> calc()0

如果已經有一個list或者tuple,要調用一個可變參數怎麼辦?可以這樣做:

>>> nums = [1, 2, 3]>>> calc(nums[0], nums[1], nums[2])14

這種寫法當然是可行的,問題是太繁瑣,所以Python允許你在list或tuple前面加一個*號,把list或tuple的元素變成可變參數傳進去:

>>> nums = [1, 2, 3]>>> calc(*nums)14

這種寫法相當有用,而且很常見。
關鍵字參數

可變參數允許你傳入0個或任意個參數,這些可變參數在函數調用時自動組裝為一個tuple。而關鍵字參數允許你傳入0個或任意個含參數名的參數,這些關鍵字參數在函數內部自動組裝為一個dict。請看樣本:

def person(name, age, **kw):  print 'name:', name, 'age:', age, 'other:', kw

函數person除了必選參數name和age外,還接受關鍵字參數kw。在調用該函數時,可以只傳入必選參數:

>>> person('Michael', 30)name: Michael age: 30 other: {}

也可以傳入任意個數的關鍵字參數:

>>> person('Bob', 35, city='Beijing')name: Bob age: 35 other: {'city': 'Beijing'}>>> person('Adam', 45, gender='M', job='Engineer')name: Adam age: 45 other: {'gender': 'M', 'job': 'Engineer'}

關鍵字參數有什麼用?它可以擴充函數的功能。比如,在person函數裡,我們保證能接收到name和age這兩個參數,但是,如果調用者願意提供更多的參數,我們也能收到。試想你正在做一個使用者註冊的功能,除了使用者名稱和年齡是必填項外,其他都是可選項,利用關鍵字參數來定義這個函數就能滿足註冊的需求。

和可變參數類似,也可以先組裝出一個dict,然後,把該dict轉換為關鍵字參數傳進去:

>>> kw = {'city': 'Beijing', 'job': 'Engineer'}>>> person('Jack', 24, city=kw['city'], job=kw['job'])name: Jack age: 24 other: {'city': 'Beijing', 'job': 'Engineer'}

當然,上面複雜的調用可以用簡化的寫法:

>>> kw = {'city': 'Beijing', 'job': 'Engineer'}>>> person('Jack', 24, **kw)name: Jack age: 24 other: {'city': 'Beijing', 'job': 'Engineer'}

參數組合

在Python中定義函數,可以用必選參數、預設參數、可變參數和關鍵字參數,這4種參數都可以一起使用,或者只用其中某些,但是請注意,參數定義的順序必須是:必選參數、預設參數、可變參數和關鍵字參數。

比如定義一個函數,包含上述4種參數:

def func(a, b, c=0, *args, **kw):  print 'a =', a, 'b =', b, 'c =', c, 'args =', args, 'kw =', kw

在函數調用的時候,Python解譯器自動按照參數位置和參數名把對應的參數傳進去。

>>> func(1, 2)a = 1 b = 2 c = 0 args = () kw = {}>>> func(1, 2, c=3)a = 1 b = 2 c = 3 args = () kw = {}>>> func(1, 2, 3, 'a', 'b')a = 1 b = 2 c = 3 args = ('a', 'b') kw = {}>>> func(1, 2, 3, 'a', 'b', x=99)a = 1 b = 2 c = 3 args = ('a', 'b') kw = {'x': 99}

最神奇的是通過一個tuple和dict,你也可以調用該函數:

>>> args = (1, 2, 3, 4)>>> kw = {'x': 99}>>> func(*args, **kw)a = 1 b = 2 c = 3 args = (4,) kw = {'x': 99}

所以,對於任意函數,都可以通過類似func(*args, **kw)的形式調用它,無論它的參數是如何定義的。
小結

Python的函數具有非常靈活的參數形態,既可以實現簡單的調用,又可以傳入非常複雜的參數。

預設參數一定要用不可變對象,如果是可變對象,運行會有邏輯錯誤!

要注意定義可變參數和關鍵字參數的文法:

  • *args是可變參數,args接收的是一個tuple;
  • **kw是關鍵字參數,kw接收的是一個dict。

以及調用函數時如何傳入可變參數和關鍵字參數的文法:

  • 可變參數既可以直接傳入:func(1, 2, 3),又可以先組裝list或tuple,再通過*args傳入:func(*(1, 2, 3));
  • 關鍵字參數既可以直接傳入:func(a=1, b=2),又可以先組裝dict,再通過**kw傳入:func(**{'a': 1, 'b': 2})。

使用*args和**kw是Python的習慣寫法,當然也可以用其他參數名,但最好使用習慣用法。

 

聯繫我們

該頁面正文內容均來源於網絡整理,並不代表阿里雲官方的觀點,該頁面所提到的產品和服務也與阿里云無關,如果該頁面內容對您造成了困擾,歡迎寫郵件給我們,收到郵件我們將在5個工作日內處理。

如果您發現本社區中有涉嫌抄襲的內容,歡迎發送郵件至: info-contact@alibabacloud.com 進行舉報並提供相關證據,工作人員會在 5 個工作天內聯絡您,一經查實,本站將立刻刪除涉嫌侵權內容。

A Free Trial That Lets You Build Big!

Start building with 50+ products and up to 12 months usage for Elastic Compute Service

  • Sales Support

    1 on 1 presale consultation

  • After-Sales Support

    24/7 Technical Support 6 Free Tickets per Quarter Faster Response

  • Alibaba Cloud offers highly flexible support services tailored to meet your exact needs.