函数式编程

核能气质少年 提交于 2020-02-09 04:15:22

函数也是对象

python中函数也是一个对象,因此可以当做参数传递

函数map()

map()函数接收两个参数,一个是函数,一个是Iterable,map将传入的函数依次作用到序列的每个元素,并把结果作为新的Iterator返回,和java1.8的map一致,用来转换数据,生成新的惰性序列

def f(x):
    return x * x
#构建一个返回x * x列表的迭代器 
r=map(f, [1,2,3,4,5,6,7,8,9])

函数reduce()

reduce把一个函数作用在一个序列[x1, x2, x3, ...]上,这个函数必须接收两个参数,reduce把结果继续和序列的下一个元素做累积计算

#求和
from functools import reduce
def add(x, y):
    return x + y

reduce(add, [1, 3, 5, 7, 9])

函数filter()

和map()类似,filter()也接收一个函数和一个序列。和map()不同的是,filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素

#过滤列表中的偶数,只保留奇数
def is_odd(n):
    return n % 2 == 1

list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
#求素数
def buildList():
    n = 1
    while True:
        n = n + 2
        yield n

def _not_divisible(n):
    return lambda x: x % n > 0

def primes():
    yield 2
    it = buildList() # 初始序列
    while True:
        n = next(it) # 返回序列的第一个数
        yield n
        it = filter(_not_divisible(n), it) # 构造新序列

for n in primes():
    if n < 1000:
        print(n)
    else:
        break

排序函数sorted()

sorted()函数也是一个高阶函数,它还可以接收一个key函数来实现自定义的排序

#按绝对值大小排序,key为函数
sorted([36, 5, -12, 9, -21], key=abs)

返回函数

把函数作为结果值返回,内部函数可以引用外部函数的参数和局部变量

#返回求和函数
def lazy_sum(*args):
    def sum():
        ax = 0
        for n in args:
            ax = ax + n
        return ax
    return sum
#此处f是返回的求和函数,非求和结果
f = lazy_sum(1, 3, 5, 7, 9)
#返回求和结果
f()

闭包

返回的函数在其定义内部引用了局部变量args,所以,当一个函数返回了一个函数后,其内部的局部变量还被新函数引用
返回函数不要引用任何循环变量,或者后续会发生变化的变量

def count():
    fs = []
    for i in range(1, 4):
        def f():
             return i*i
        fs.append(f)
    return fs

f1, f2, f3 = count()
print(f1())
print(f2())
print(f3())

结果

9
9
9

lambda写法

def count():
    def f(j):
        return lambda: j*j
    fs = []
    for i in range(1, 4):
        fs.append(f(i))
    return fs

f1, f2, f3 = count()
print(f1())
print(f2())
print(f3())

匿名函数

def f(x):
    return x * x

等价于

f = lambda x: x * x

装饰器

函数对象有一个__name__属性,可以拿到函数的名字,假设我们要增强now()函数的功能,比如,在函数调用前后自动打印日志,但又不希望修改now()函数的定义,这种在代码运行期间动态增加功能的方式,称之为“装饰器”(Decorator)
本质上,decorator就是一个返回函数的高阶函数,python从语法层次支持decorator,也支持OOP的decorator,个人认为语言级别的使用起来更加简单方面

  • 两层嵌套
def log(func):
    def wrapper(*args, **kw):
        print('call %s():' % func.__name__)
        return func(*args, **kw)
    return wrapper

因为它是一个decorator,所以接受一个函数作为参数,并返回一个函数
使用:

@log
def now():
    print('2015-3-25')

调用now()函数,不仅会运行now()函数本身,还会在运行now()函数前打印一行日志
把@log放到now()函数的定义处,相当于执行了语句:

now = log(now)
  • 三层嵌套
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('2015-3-25')

相当于:

now = log('execute')(now)
  • 修改装饰器的签名
import functools

def log(func):
    @functools.wraps(func)
    def wrapper(*args, **kw):
        print('call %s():' % func.__name__)
        return func(*args, **kw)
    return wrapper

偏函数

当函数的参数个数太多,需要简化时,使用functools.partial可以创建一个新的函数,这个新函数可以固定住原函数的部分参数,从而在调用时更简单。

#构建把一个数转成2进制的函数
import functools
int2 = functools.partial(int, base=2)

调用

int2(7)
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!