python中实现函数不限制参数的数量

我们在写Python代码的时候，有的时候需要设计一种函数，这种函数的参数个数不确定，或者说要求函数的参数不限个数，只有在函数被调用的时候，才能确定参数的数量。本文介绍如何实现这样的函数参数接口的设计。

tuple参数(*args)

我们在设计函数参数的时候，如果使用一个*符号后跟字母，表示这是一个tuple参数，tuple内数据的个数，就不限制了。

我们来设计一个简单的函数，这个函数的作用就是将tuple参数内所有的数据相加求和：

>>> def add(*args):
...     print(args)
...     print(type(args))
...     sum = 0
...     for i in range(len(args)):
...         sum += args[i]
...     print(sum)
...
>>> add(1,2,3)
(1, 2, 3)
<class 'tuple'>
6
>>> add(1,2,3,8,9,666)
(1, 2, 3, 8, 9, 666)
<class 'tuple'>
689
>>> add()
()
<class 'tuple'>
0

add函数的参数，args，符号后面必须要有字母跟随，如果参数是一个独立的*符号，语义是不一样的！

args参数在函数内部，是一个tuple类型的对象，使用时要去掉前面的符号，直接使用args。在调用函数时，看不出来这是一个tuple，调用时并无特别之处，只是参数的数量不确定，可以很多，注意哦，也可以没有参数。

实现不限参数个数的函数接口，一般都使用这种方式，tuple对象内的各数据类型应该保持一致，如果不一致，大概率这不是一个好的设计，因为对各数据的处理方式会不一样。

dict参数（**kwargs）

设计函数参数时，如果使用两个**符号后跟字母，表示这个参数是一个dict对象。这种方式也可实现函数参数的个数不限。

下面的代码示例，是一个输入dict参数的函数，函数的作用，将dict对象中的数据，全部以 key : value 配对的方式显示出来：

>>> def test_kw(**kwargs):
...     print(kwargs)
...     print(type(kwargs))
...     for k,v in kwargs.items():
...         print(k,':',v)
...
>>> test_kw(a=1,b=2,c=3,d=4)
{'a': 1, 'b': 2, 'c': 3, 'd': 4}
<class 'dict'>
a : 1
b : 2
c : 3
d : 4
>>> test_kw(a=1,b=2,c=3,d=4,x=999)
{'a': 1, 'b': 2, 'c': 3, 'd': 4, 'x': 999}
<class 'dict'>
a : 1
b : 2
c : 3
d : 4
x : 999
>>> test_kw()
{}
<class 'dict'>

采用**kwargs参数，函数调用时，要写成key=value的形式。同样的，这种方式也兼容没有参数输入。

使用这种方式的dict参数，函数内部可能需要预先知道有哪些key值可以使用，或者就是直接遍历所有的key和value。我想到了这样一个应用场景，设计一个函数，作用是替换输入字符串中的某些字符，返回新字符串，替换规则通过**kwargs传入。代码实现如下：

>>> def char_replace(originstr, **kwargs):
...     newstr = ''
...     for c in originstr:
...         if c in kwargs.keys():
...             newstr += kwargs[c]
...         else:
...             newstr += c
...     return newstr
...
>>>
>>> char_replace('abcde', a='x')
'xbcde'
>>> char_replace('abcde', a='x',c='y')
'xbyde'
>>> char_replace('abcde', a='x',c='y',e='z')
'xbydz'

以上就是在Python中，实现函数参数个数不确定的两种方法。