Python: 函数参数魔法
2016-12-01
函数参数
在 Python 中,定义函数和调用函数都很简单,但如何定义函数参数和传递函数参数,则涉及到一些套路了。总的来说,Python 的函数参数主要分为以下几种:
必选参数
默认参数
可变参数
关键字参数
必选参数
必选参数可以说是最常见的了,顾名思义,必选参数就是在调用函数的时候要传入数量一致的参数,比如:
1
2
3
4
5
6
7
8
9
10
11
12
13
>>> def add(x, y): # x, y 是必选参数
... print x + y
...
>>> add() # 啥都没传,不行
Traceback (most recent call last):
File "", line 1, in
TypeError: add() takes exactly 2 arguments (0 given)
>>> add(1) # 只传了一个,也不行
Traceback (most recent call last):
File "", line 1, in
TypeError: add() takes exactly 2 arguments (1 given)
>>> add(1, 2) # 数量一致,通过
3
默认参数
默认参数是指在定义函数的时候提供一些默认值,如果在调用函数的时候没有传递该参数,则自动使用默认值,否则使用传递时该参数的值。
看看例子就明白了:
1
2
3
4
5
6
7
>>> def add(x, y, z=1): # x, y 是必选参数,z 是默认参数,默认值是 1
... print x + y + z
...
>>> add(1, 2, 3) # 1+2+3
6
>>> add(1, 2) # 没有传递 z,自动使用 z=1,即 1+2+1
4
可以看到,默认参数使用起来也很简单,但有两点需要注意的是:
默认参数要放在所有必选参数的后面
默认参数应该使用不可变对象
比如,下面对默认参数的使用是错误的:
1
2
3
4
5
6
7
8
9
10
11
>>> def add(x=1, y, z): # x 是默认参数,必须放在所有必选参数的后面
... return x + y + z
...
File "", line 1
SyntaxError: non-default argument follows default argument
>>>
>>> def add(x, y=1, z): # y 是默认参数,必须放在所有必选参数的后面
... return x + y + z
...
File "", line 1
SyntaxError: non-default argument follows default argument
再来看看为什么默认参数应该使用不可变对象。
我们看一个例子:
1
2
3
>>> def add_to_list(L=[]):
... L.append('END')
... return L
在上面的函数中,L 是一个默认参数,默认值是 [],表示空列表。
我们来看看使用:
1
2
3
4
5
6
7
8
9
10
>>> add_to_list([1, 2, 3]) # 没啥问题
[1, 2, 3, 'END']
>>> add_to_list(['a', 'b', 'c']) # 没啥问题
['a', 'b', 'c', 'END']
>>> add_to_list() # 没有传递参数,使用默认值,也没啥问题
['END']
>>> add_to_list() # 没有传递参数,使用默认值,竟出现两个 'END'
['END', 'END']
>>> add_to_list() # 糟糕了,三个 'END'
['END', 'END', 'END']
为啥呢?我们在调用函数的时候没有传递参数,那么就默认使用 L=[],经过处理,L 应该只有一个元素,怎么会出现调用函数两次,L 就有两个元素呢?
原来,L 指向了可变对象 [],当你调用函数时,L 的内容发生了改变,默认参数的内容也会跟着变,也就是,当你第一次调用时,L 的初始值是 [],当你第二次调用时,L 的初始值是 ['END'],等等。
所以,为了避免不必要的错误,我们应该使用不可变对象作为函数的默认参数。
可变参数
在某些情况下,我们在定义函数的时候,无法预估函数应该制定多少个参数,这时我们就可以使用可变参数了,也就是,函数的参数个数是不确定的。
看看例子:
1
2
3
4
5
6
>>> def add(*numbers):
... sum = 0
... for i in numbers:
... sum += i
... print 'numbers:', numbers
... return sum
在上面的代码中,numbers 就是一个可变参数,参数前面有一个 * 号,表示是可变的。在函数内部,参数 numbers 接收到的是一个 tuple。
在调用函数时,我们可以给该函数传递任意个参数,包括 0 个参数:
1
2
3
4
5
6
7
8
9
10
11
12
>>> add() # 传递 0 个参数
numbers: ()
0
>>> add(1) # 传递 1 个参数
numbers: (1,)
1
>>> add(1, 2) # 传递 2 个参数
numbers: (1, 2)
3
>>> add(1, 2, 3) # 传递 3 个参数
numbers: (1, 2, 3)
6
上面的 * 表示任意参数,实际上,它还有另外一个用法:用来给函数传递参数。
看看例子:
1
2
3
4
5
6
7
8
9
10
11
>>> def add(x, y, z): # 有 3 个必选参数
... return x + y + z
...
>>> a = [1, 2, 3]
>>> add(a[0], a[1], a[2]) # 这样传递参数很累赘
6
>>> add(*a) # 使用 *a,相当于上面的做法
6
>>> b = (4, 5, 6)
>>> add(*b) # 对元组一样适用
15
再看一个例子:
1
2
3
4
5
6