[toc]
什么是函数式编程?
函数式编程是一种抽象程度很高的编程范式,纯粹的函数式编程语言编写的函数没有变量,因此,任意一个函数,只要输入是确定的,输出就是确定的,这种纯函数我们称之为没有副作用。
而允许使用变量的程序设计语言,由于函数内部的变量状态不确定,同样的输入,可能得到不同的输出,因此,这种函数是有副作用的
函数式编程的一个特点就是,允许把函数本身作为参数传入另一个函数,还允许返回一个函数!
Python对函数式编程提供部分支持。由于Python允许使用变量,因此,Python不是纯函数式编程语言
高阶函数
Python中函数两个特性
变量可以指向函数
函数本身可以赋值给变量,通过变量可以调用函数
>>> f = abs
>>> f(-10)
10
函数名是变量
把函数名看作变量,可将其指向其他对象,则无法调用原函数
>>> abs = 10
>>> abs(-10)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'int' object is not callable
传入函数
由于变量可以指向函数,函数的参数能接收变量,那么一个函数就可以接收另一个函数作为参数,这种函数就称之为高阶函数
一个最简单的高阶函数:
def add(x, y, f)
return f(x) + f(y)
当调用add(-5, -6, abs)
时,函数计算abs(-5)+abs(-6)
,返回11
把函数作为参数传入,这样的函数称为高阶函数,函数式编程就是指这种高度抽象的编程范式
Python内建高阶函数
map()函数
|map()
函数接收两个参数,一个是函数,一个是Iterable
,map
将传入的函数依次作用到序列的每个元素,并把结果作为新的Iterator
返回
def f(x):
return x * x
>>> r = map(f, [1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> list(r)
[1, 4, 9, 16, 25, 36, 49, 64, 81]
reduce()函数
|reduce
把一个函数作用在一个序列[x1, x2, x3, ...]
上,这个函数必须接收两个参数,reduce
把结果继续和序列的下一个元素做累积计算,其效果就是:
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)
filter()函数
该函用于过滤序列。
|filter()
接收一个函数和一个序列,filter()
把传入的函数依次作用于每个元素,然后根据返回值是True
还时False
决定保留还是丢弃该元素。
例如,删掉list中的偶数,保留奇数
def is_odd(n):
return n % 2 == 1
>>> list(filter(is_odd, [1, 2, 3, 4, 5]))
[1, 3, 5]
用filter求素数
利用埃氏筛法
- 先构造一个从
3
开始的奇数序列
利用生成器
def _odd_iter():
n = 1
while 1:
n += 2
yield n
- 定义一个筛选函数
def _not_divisible(n):
return lambda x: x % n > 0
- 最后定义一个生成器,不断返回下一素数
def primes():
yield 2
it = _odd_iter()
while 1:
n = next(it)
yield n
it = filter(_not_divisible(n), it)
- 调用得到1000内素数
for n in primes():
if (n < 1000):
print(n)
else:
break
|filter()
的作用是从一个序列中筛出符合条件的元素。由于filter()
使用了惰性计算,所以只有在取filter()
结果的时候,才会真正筛选并每次返回下一个筛出的元素。
sorted()函数
它可以接收一个key
函数来实现自定义的排序,例如按绝对值大小排序:
>>> sorted([36, 5, -12, 9, -21], key=abs)
[5, 9, -12, -21, 36]
key指定的函数将作用于list的每一个元素上,并根据key函数返回的结果进行排序。
然后sorted()
函数按照keys进行排序,并按照对应关系返回list相应的元素
返回函数&闭包
函数作为返回值
def lazy_f(x, y):
def f():
return x + y
return f1 #返回函数
>>> f1 = lazy_f(1, 2)
>>> f2 = lazy_f(1, 2)
>>> f1 == f2
False #两次返回的函数并不同
>>> f1() #调用函数时,才真正计算求和结果
3
>>> f2()
3
闭包
当lazy_f
返回函数f
时,相关参数和变量都保存在返回的函数中,这种称为“闭包(Closure)”的程序结构拥有极大的威力。
需要注意的是,返回的函数并没有立刻执行,而是直到调用了f()
才执行
如
def count():
fs = []
for i in range(1, 4):
def f():
return i*i
fs.append(f)
return fs
f1, f2, f3 = count()
在实际调用中
>>> f1()
9
>>> f2()
9
>>> f3()
9
注意:返回函数不要引用任何循环变量,或者后续会发生变化的变量
匿名函数
如匿名函数lambda x: x * x
关键字lambda
表示匿名函数,冒号前面的x
表示函数参数。
匿名函数只能有一个表达式,不写return
,返回值就是该表达式的结果
Python对匿名函数的支持有限,只有一些简单的情况下可以使用匿名函数。
装饰器
函数对像有一个__name__
属性,可以返回函数的名字
>>> now.__name__
'now'
现在,假设我们要增强now()
函数的功能,比如,在函数调用前后自动打印日志,但又不希望修改now()
函数的定义,这种在代码运行期间动态增加功能的方式,称之为“装饰器(Decorator)”。
定义一个能打印日志的decorator
def log(func):
def wrapper(*args, **kw): #*args是非关键字参数,用于元组,**kw是关键字参数,用于字典
print('class %s():' % func.__name__)
return func(*args, **kw)
return wrapper
借助Python的@语法,把decorator置于函数的定义处:
@log
def now():
print('2019/6/21')
调用函数now()
,不仅会运行now()
本身,还会在其之前打印一行日志
>>> now()
call now():
2019/6/21
相当于执行了语句:
now = log(now)
如果decorator本身需要传入参数,那就需要编写一个返回decorator的高阶函数,写出来会更复杂。比如,要自定义log的文本:
def log(text):
def decorator(func):
def wrapper(*args, **kw):
print('%s %s():' % (text, func.__name__))
return func(*args, **kw)
return wrapper
return decorator
使用
@log('execute')
def now():
print('2019/6/21')
#相当于now = log('excute')(now)
以上两种decorator的定义都没有问题,但还差最后一步。因为我们讲了函数也是对象,它有__name__
等属性,但你去看经过decorator装饰之后的函数,它们的__name__
已经从原来的'now'
变成了'wrapper'
:
>>> now.__name__
'wrapper'
因为返回的那个wrapper()
函数名字就是'wrapper'
,所以,需要把原始函数的__name__
等属性复制到wrapper()
函数中,否则,有些依赖函数签名的代码执行就会出错。
不需要编写wrapper.__name__ = func.__name__
这样的代码,Python内置的functools.wraps
就是干这个事的,所以,一个完整的decorator的写法如下:
import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('call %s():' % func.__name__)
return func(*args, **kw)
return wrapper
或者针对带参数的decorator:
import functools
def log(text):
def decorator(func):
@functools.wraps(func)
def wrapper(*args, **kw):
print('%s %s():' % (text, func.__name__))
return func(*args, **kw)
return wrapper
return decorator
偏函数
可借助functools.partial
创建偏函数:把一个函数的某些参数设为某个默认值,返回一个新的函数,调用这个新函数会更简单
>>> int2 = functools.partial(int, base = 2)
>>> int2('10000')
16
当函数的参数个数太多,需要简化时,使用
functools.partial
可以创建一个新的函数,这个新函数可以固定住原函数的部分参数,从而在调用时更简单。