【Python技巧】巧妙地使用参数来缓存重复计算的数据

先举一个很简单的例子, 一个函数接收两个参数,返回这两个参数的和。

def add(a, b):
    return a + b

def main():
    print(add(1, 2))
    print(add(3, 4))
    print(add(1, 2))
    print(add(3, 4))

if __name__ == '__main__':
    main()

我们对1和2,3和4分别进行了两次计算,那么add函数每次都会进行计算,然后返回值,如果可以发现1和2被计算过了,以后再调用1和2求和的话,能直接使用计算好的值是最好的,而不用重新计算(如果计算过程复杂耗时较长的情况下,这是非常有用的方式,这里只是简单的进行了求和计算,计算量较少)。

看一下我们的改进方式:

def add(a, b, __cache={}):
    if (a, b) in __cache:
        return __cache[(a, b)]

    ret = a + b
    __cache[(a, b)] = ret
    return ret


def main():
    print(add(1, 2))
    print(add(3, 4))
    print(add(1, 2))
    print(add(3, 4))


if __name__ == '__main__':
    main()

我们对add函数增加了一个字典参数,大家都知道,任何调用如果对这个字典进行了修改,这个字典会一直有效的。
比如第一次调用add(1, 2)的时候:

if (a, b) in __cache:
        return __cache[(a, b)]

(a, b)也就是1和2组成的元组这个key不存在于字典__cache中,那么会执行后面的计算,并且把计算结果以字典的方式存在__cache中,其中key是参数组成的元组,value是计算后的值,然后返回计算后的值,同理3和4的计算类似,那下一次再遇见add(1, 2)的时候:

if (a, b) in __cache:
        return __cache[(a, b)]

发现(a, b)是存在于__cache中的,我们直接返回已经计算好的值即可,而不用重新计算(如果计算量较大,这会非常好用)。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容