- 可变参数
我们先来看看俩个参数的函数形式:
# 两个数相加
def sum(num1, num2):
return num1 + num2
print sum(1,2)
现在,假设不知道要传入参数的个数,只是想把传入的参数都累加起来,那么可以这样定义函数:
def sum(numbers):
total = 0
for i in numbers:
total += i
return total
print sum([1,2,3,4])
print sum([3,4,5])
没错,在定义时只写一个参数就行了。实际上我们传入的是一个list,这从某种程度上实现了我们要的要求.
对于传入list这里有一个巨坑,如下代码:
def addNumber(num=[]):
num.append(0)
return num
print addNumber([1,2])
print addNumber([1,2,3])
输出:
[1, 2, 0]
[1, 2, 3, 0]
哎,看上去好像没什么问题啊,但是,如执行下列代码:
print addNumber()
print addNumber()
输出:
[0]
[0, 0]
因为Python函数在定义的时候,默认参数num的值就被计算出来了,即[],因为默认参数num也是一个变量,它指向对象[],每次调用该函数,如果改变了num的内容,则下次调用时,默认参数的内容就变了,不再是函数定义时的[]了。
所以,定义默认参数要牢记一点:默认参数必须指向不变对象!
改进后代码:
def addNumber(num=None):
if num is None:
num = []
num.append(0)
return num
print addNumber()
print addNumber()
[0]
[0]
哎成功了。接下来,看Python 的另一种实现:
def sum(*num):
total = 0
for i in num:
total += i
return total
print sum(1,2,3,4)
print sum(1,2)