Python中的函数
定义函数
在Python中,定义一个函数要使用def语句,依次写出函数名、括号、括号中的参数和冒号:,然后,在缩进块中编写函数体,函数的返回值用return语句返回。
我们以自定义一个求绝对值的my_abs函数为例:
def my_abs(x):
if x>=0:
return x
else:
return -x
print my_abs(-12)
# 输出:12
如果没有return语句,函数执行完毕后也会返回结果,只是结果为None。
空函数
如果想定义一个什么事也不做的空函数,可以用pass语句:
def nop():
pass
pass语句什么都不做,那有什么用?实际上pass可以用来作为占位符,比如现在还没想好怎么写函数的代码,就可以先放一个pass,让代码能运行起来。
参数检查
调用函数时,如果参数个数不对,Python解释器会自动检查出来,并抛出TypeError:
def my_abs(x):
if x>=0:
return x
else:
return -x
print my_abs(-12,2)
"""
报错:
Traceback (most recent call last):
File "G:/python/day2/my_abs.py", line 7, in <module>
print my_abs(-12,2)
TypeError: my_abs() takes exactly 1 argument (2 given)
"""
但是如果参数类型不对,Python解释器就无法帮我们检查
实现参数类型检查可如下做法:
def my_abs(x):
if not isinstance(x, (int, float)):
raise TypeError('bad operand type')
if x >= 0:
return x
else:
return -x
返回多个值
import math
def move(x, y, step, angle=0):
nx = x + step * math.cos(angle)
ny = x - step * math.sin(angle)
return nx, ny
x, y = move(100, 100, 60, math.pi/6)
print x, y
# 输出:151.961524227 70.0
但其实这只是一种假象,Python函数返回的仍然是单一值:
import math
def move(x, y, step, angle=0):
nx = x + step * math.cos(angle)
ny = x - step * math.sin(angle)
return nx, ny
r = move(100, 100, 60, math.pi/6)
print r
# 输出:(151.96152422706632, 70.0)
由输出结果可明显看出,返回值是一个tuple(元组),在语法上,返回一个tuple可以省略括号,而多个变量可以同时接收一个tuple,按位置赋给对应的值,所以,Python的函数返回多值其实就是返回一个tuple,但写起来更方便。
函数的参数
默认参数
def enroll(name, gender, age=6, city='Beijing'):
print 'name:', name
print 'gender:', gender
print 'age:', age
print 'city:', city
默认参数降低了参数的参与度,而一旦需要更复杂的调用时,又可以传递更多的参数来实现。无论是简单调用还是复杂调用,函数只需要定义一个。
注意:默认参数的坑
def add_end(L=[]):
L.append('END')
return L
add_end()
# 输出:['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
可变参数
def calc(*numbers):
sum = 0
for n in numbers:
sum = sum + n * n
return sum
calc(1, 2)
# 输出:5
calc()
# 输出:0
如果已经有一个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('csx', 30)
# 输出:name: csx 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这两个参数,但是,如果调用者愿意提供更多的参数,我们也能收到。试想你正在做一个用户注册的功能,除了用户名和年龄是必填项外,其他都是可选项,利用关键字参数来定义这个函数就能满足注册的需求。
简化写法
kw = {'city': 'Beijing', 'job': 'Engineer'}
person('Jack', 24, **kw)
# 输出:name: Jack age: 24 other: {'city': 'Beijing', 'job': 'Engineer'}
参数组合
在Python中定义函数,可以用必选参数、默认参数、可变参数和关键字参数,这4种参数都可以一起使用,或者只用其中某些,但是请注意,参数定义的顺序必须是:必选参数、默认参数、可变参数和关键字参数。