Python實戰從入門到精通第十一講——可接受任意數量參數的函數

構造一個可接受任意數量參數的函數。

爲了能讓一個函數接受任意數量的位置參數，可以使用一個*參數。例如：

def avg(first, *rest):
    return (first + sum(rest)) / (1 + len(rest))

# Sample use
avg(1, 2) # 1.5
avg(1, 2, 3, 4) # 2.5

這個例子中，rest是由所有其他位置參數組成的元組。然後我們在代碼中把它當成了一個序列來進行後續的計算。

爲了接受任意數量的關鍵字參數，使用一個以**開頭的參數。比如：

import html

def make_element(name, value, **attrs):
    keyvals = [' %s="%s"' % item for item in attrs.items()]
    attr_str = ''.join(keyvals)
    element = '<{name}{attrs}>{value}</{name}>'.format(
                name=name,
                attrs=attr_str,
                value=html.escape(value))
    return element

# Example
# Creates '<item size="large" quantity="6">Albatross</item>'
make_element('item', 'Albatross', size='large', quantity=6)

# Creates '<p>&lt;spam&gt;</p>'
make_element('p', '<spam>')

在這裏，attrs是一個包含所有被傳入進來的關鍵字參數的字典。

如果你還希望某個函數能同時接受任意數量的位置參數和關鍵字參數，可以同時使用*和**。比如：

def anyargs(*args, **kwargs):
    print(args) # A tuple
    print(kwargs) # A dict

使用這個函數時，所有位置參數會被放到args元組中，所有關鍵字參數會被放到字典kwargs中。

一個*參數只能出現在函數定義中最後一個位置參數後面，而 **參數只能出現在最後一個參數。 有一點要注意的是，在*參數後面仍然可以定義其他參數。

def a(x, *args, y):
    pass

def b(x, *args, y, **kwargs):
    pass