zoukankan      html  css  js  c++  java
  • 如何使用函数装饰器?

    需求:
    某些时候我们想为多个函数,统一添加某种功能,比如计时统计、日志记录、缓存运算结果等等
    我们不想在每个函数内一一添加完全相同的代码,有什么好的解决方案?

    思路:
    定义装饰器函数,用它来生成一个在原函数基础上添加了新功能的函数,替代原函数

    代码:

    def memo(func):
        cache = {}
        def wrap(*args):
            res = cache.get(args) # 此处args传入的是一个元组,元组可以作为字典的键
            if not res:
                res = cache[args] = func(*args) # 将传入的元组参数打散来
            return res
        return wrap
    
    # [题目1] 斐波那契数列(Fibonacci sequence):
    # F(0)=1,F(1)=1,F(n)=F(n-1)+F(n-2)(n>=2)
    # 1,1,2,3,5,8,13,21,34,...
    # 求数列第n项的值
    # 使用缓存,可以去除重复的大量运算,提高运算的速度
    @memo # 等价于fibonacci = memo(fibonacci)
    def fibonacci(n):
        if n <= 1:
            return 1
        res = fibonacci(n-1) + fibonacci(n-2)
        return res
    
    
    # [题目2] 走楼梯问题
    # 有100阶楼梯,一个人可以迈1~3阶,一共有多少走法?
    @memo
    def climb(n,steps):
        count = 0
        if n == 0:
            count = 1
        elif n > 0:
            for step in steps:
                count += climb(n-step,steps)
        return count
    
    # fibonacci = memo(fibonacci)  # 偷梁换柱,用户直接调用的还是fibonacci,不感知函数内部
    print(fibonacci(50))
    
    # climb = memo(climb)  # 偷梁换柱,用户直接调用的还是climb,不感知函数内部
    print(climb(100,(1,2,3)))
    
  • 相关阅读:
    Mac上安装使用Nginx
    突然想到了王自如
    sqls
    spring利用后置处理器初始化bean属性
    replace的用法
    Redis使用记录
    netty handlers模式
    webmagic 下载页面
    webmagic 初始化 startRequests
    Singleton模式
  • 原文地址:https://www.cnblogs.com/Richardo-M-Q/p/13957363.html
Copyright © 2011-2022 走看看