python学习笔记之函数
在了解了python的基本数据类型后，现在我们来看一下python中的函数，为了便于维护和更好的实现模块化，程序都会分解为多个函数。下面我将总结一下python函数的特性，装饰器，生成器，协程等，他们都是进行函数编程和数据处理的强大工具。
1.函数的定义
在python中定义函数很简单，使用def关键字。和c++一样python的函数也可以使用默认参数。
|
1
2
3
4
|
defadd(x, y=1):
returnx+y
>>> add(1)
2
|
python也可以接收任意数量的参数，在最后一个变量前面加星号，python就会把剩下的变量作为元组放入这个变量。
|
1
2
3
4
5
6
|
defadd(x, y=1,*args):
foriinargs:
x+=i
returnx+y
>>> add(1,2,3,4,5)
15
|
上面传递的参数都是根据参数的位置赋值给相应的变量，称为位置参数；python还支持显式地参数赋值，称为关键字参数。
|
1
2
3
4
|
defadd(x, y, z, w):
printx,y,z,w
>>> add(1, z=2, w=3, y=4)
1423
|
注意事项：
- 关键字参数的顺序无关紧要。
- 位置参数和关键字参数可以一起使用，但关键字参数必须在位置参数后面。
- 位置参数已经赋值的变量不能再使用关键字参数赋值，比如上面的x已经赋值为1了，后面的关键字参数就不可以再对x赋值。
参数前一个星号表示任意数量的位置参数，两个星号就是任意数量的关键字参数。python会把剩下的关键字参数作为字典放入这个变量。
|
1
2
3
4
|
deffoo(*args,**kwargs):
printargs,kwargs
>>> foo(1,2,3,4, x=5, y=6)
(1,2,3,4) {'y':6,'x':5}
|
2.装饰器
装饰器是一个函数，主要用于包装另一个函数或类，以修改或增强这个函数或类的行为。装饰器使用@符号表示。
|
1
2
3
4
5
6
7
8
9
10
11
12
|
deftrace(func):
print'trace'
returnfunc
@trace
deffoo():
print'foo'
foo()
#输出
trace
foo
|
第一个函数trace()是装饰器函数，他接收一个函数对象func，并返回函数对象(必须返回，但不一定是func)，@trace对函数foo()进行修饰，等价于foo = trace(foo)，所以最后调用的foo并不是之前定义的foo()，而是trace(foo)。这个例子是在函数foo调用前做一些修饰，也可以在调用前后做一些事情。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
# 修改trace
deftrace(func):
print'trace'
defnewFunc():
print'begin'
func()
print'end'
returnnewFunc
# 运行foo后输出
trace
begin
foo
end
|
装饰器是函数，当然也可以接收参数。这样的话必须在装饰器函数内部再定义一个新装饰器函数来接收func对象。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
deftrace(msg):
print'trace'
printmsg
defnewTrace(func):
defnewFunc():
print'begin'
func()
print'end'
returnnewFunc
returnnewTrace
@trace('message')
deffoo():
print'foo'
foo()
# 输出
trace
message
begin
foo
end
|
3.生成器
生成器是一个函数，他生成一个值的序列，用于迭代。生成器使用yield关键字定义。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
>>>defcountdown(n):
whilen>0:
yieldn
n-=1
>>> c=countdown(5)
>>> c.next()
5
>>>next(c)
4
>>>foriinc:
printi
321
|
上面的c就是一个生成器对象，他可以进行迭代来历遍，在countdown函数内遇到yield就会中止并返回n，直到下次调用next。
3.1.生成器表达式
生成器表达式可以生产一个简单的生成器对象。他和for循环类似，使用圆括号，语法为：
(expression for item in iterable if condition) 后面的if可以省略。
|
1
2
3
4
|
>>> a=(2*xforxinxrange(20)ifx%3==0)
>>>foriina:
printi
061218243036
|
3.2.列表推导表达式
这里再说一下列表推导表达式，他的语法和生成器表达式差不多，使用中括号，他将生产一个新的列表，语法为：
[expression for item in iterable if condition] if同样可以省略
|
1
2
|
>>> [2*xforxinxrange(20)ifx%3==0]
[0,6,12,18,24,30,36]
|
4.协程
协程与生成器类似，只不过他是接收数据，而生成器是输出数据。同样使用yield关键字，作为右值。
|
1
2
3
4
5
6
7
8
9
10
|
>>>defreceiver():
whileTrue:
x=(yield)
print'Got ',x
>>> r=receiver()
>>> r.next()
>>> r.send(12)
Got 12
>>> r.send('wuyuans.com')
Got wuyuans.com
|
协程在使用时必须先next，让程序运行到yield关键字。这时我们可以给他加个装饰器，让他自动运行一次next。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
|
defcoroutine(func):
defstart(*args,**kwargs):
g=func(*args,**kwargs)
g.next()
returng
returnstart
@coroutine
defreceiver():
whileTrue:
x=(yield)
print'Got ',x
r=receiver()
r.send(12)
# 输出
Got 12
|