0

0

Python函数如何用 functools 模块优化函数 Python函数 functools 工具的基础应用技巧​

星夢妙者

星夢妙者

发布时间:2025-08-15 17:51:01

|

398人浏览过

|

来源于php中文网

原创

functools模块通过提供lru_cache、partial和wraps等工具显著提升python代码效率与质量。1. lru_cache通过缓存函数结果避免重复计算,尤其适用于参数不变的高耗时函数,如远程数据获取或递归算法,能大幅提升性能,但要求参数可哈希且需注意内存占用;2. partial通过固定函数部分参数生成新函数,实现逻辑复用与调用简化,相比lambda更具可读性和可维护性,适用于创建专用函数变体或与高阶函数结合使用;3. wraps用于装饰自定义装饰器中的包装函数,确保被装饰函数的__name__、__doc__等元数据得以保留,避免调试困难和工具失效,是编写健壮装饰器的必要步骤。

Python函数如何用 functools 模块优化函数 Python函数 functools 工具的基础应用技巧​

functools
模块在 Python 中扮演着一个幕后英雄的角色,它提供了一系列高阶函数和装饰器,能帮助我们对函数进行包装、修改或增强,从而让代码更高效、更简洁,也更易于维护。它的核心在于提供工具来处理函数本身,而不是函数内部的逻辑。

解决方案

functools
模块的核心价值在于它提供了一些非常实用的工具,让我们能以更优雅的方式处理函数。其中最常用也最基础的,包括
lru_cache
用于性能优化、
partial
用于函数参数的灵活绑定、以及
wraps
用于编写健壮的装饰器。理解并掌握它们,能显著提升我们编写 Python 代码的效率和质量。

lru_cache
在实际开发中如何提升函数性能?

lru_cache
functools
模块里我个人觉得最“立竿见影”的一个工具。它是一个装饰器,核心功能是实现“备忘录模式”(memoization),也就是缓存函数的调用结果。想象一下,你有一个计算量很大的函数,比如一个复杂的数学运算,或者需要查询数据库、调用外部API的函数。如果这个函数经常被相同的参数调用,那么每次都重新计算一遍无疑是巨大的资源浪费。

立即学习Python免费学习笔记(深入)”;

lru_cache
的作用就在于此。当你用
@lru_cache()
装饰一个函数后,Python 会自动为这个函数维护一个缓存。每次函数被调用时,它会先检查传入的参数是否在缓存中。如果在,直接返回缓存的结果,避免了实际的计算;如果不在,则执行函数体,并将结果存入缓存,以备下次使用。这个“LRU”代表“Least Recently Used”,意味着当缓存达到最大容量时,最近最少使用的条目会被淘汰,为新的结果腾出空间。

我个人觉得,很多时候我们写一些服务端的接口,如果底层数据源不经常变动,或者某些查询条件重复率很高,

lru_cache
简直是神器,能瞬间把响应时间拉下来。比如,一个获取用户配置的函数,配置不经常变动,但查询频繁,用
lru_cache
就能极大减轻数据库压力。当然,它也有局限性,比如缓存的数据是基于函数参数的,如果参数是不可哈希的对象(如列表、字典),就不能直接用。此外,缓存的内存占用也需要考虑,如果缓存的数据量巨大,可能会适得其反。

import time
from functools import lru_cache

# 模拟一个耗时操作,比如网络请求或复杂计算
@lru_cache(maxsize=None) # maxsize=None 意味着缓存大小无限制,但通常建议设置一个上限
def fetch_data_from_remote(item_id: str) -> str:
    print(f"正在从远程获取数据 for {item_id}...")
    time.sleep(2) # 模拟网络延迟
    return f"数据 for {item_id} 已获取"

print(fetch_data_from_remote("user_profile_1")) # 第一次调用,会等待2秒
print(fetch_data_from_remote("user_profile_2")) # 第一次调用,会等待2秒
print(fetch_data_from_remote("user_profile_1")) # 第二次调用,直接从缓存返回,几乎瞬时
print(fetch_data_from_remote("user_profile_2")) # 第二次调用,直接从缓存返回,几乎瞬时

# 斐波那契数列的经典优化
@lru_cache(maxsize=128)
def fibonacci(n):
    if n <= 1:
        return n
    return fibonacci(n - 1) + fibonacci(n - 2)

print("\n计算斐波那契数列:")
print([fibonacci(n) for n in range(10)]) # 效率很高,因为重复计算被缓存了

使用
partial
实现函数参数的灵活绑定与复用

functools.partial
提供了一种非常优雅的方式来“固定”一个函数的部分参数,从而生成一个新的函数。这在很多场景下都非常有用,尤其是当你有一个通用函数,但又想基于它创建一些更具体、更专业的版本时。

编程语言Perl性能优化的三大技巧总结 中文WORD版
编程语言Perl性能优化的三大技巧总结 中文WORD版

本文和大家重点讨论一下Perl性能优化技巧,利用Perl开发一些服务应用时,有时会遇到Perl性能或资源占用的问题,可以巧用require装载模块,使用系统函数及XS化模块,自写低开销模块等来优化Perl性能。 Perl是强大的语言,是强大的工具,也是一道非常有味道的菜:-)利用很多perl的特性,可以实现一些非常有趣而实用的功能。希望本文档会给有需要的朋友带来帮助;感兴趣的朋友可以过来看看

下载

举个例子,你可能有一个通用的

send_notification(message, type, recipient)
函数。在实际应用中,你可能经常需要发送“邮件通知”或“短信通知”。与其每次都重复
send_notification("你好", "email", "user@example.com")
这样的调用,你可以用
partial
创建一个
send_email_notification
send_sms_notification

刚开始接触

partial
,可能会觉得和
lambda
有点像,但实际用起来,尤其是在需要传递给其他函数或者保持函数签名的场景,
partial
的优势就出来了,代码会更清晰,可读性也更好。
partial
创建的函数对象,其
__name__
__doc__
属性通常比
lambda
更具描述性,也更容易被内省工具识别。它也比手动封装一层函数更简洁。

from functools import partial

def power(base, exponent):
    """计算 base 的 exponent 次幂。"""
    return base ** exponent

# 创建一个专门计算平方的函数
square = partial(power, exponent=2)
print(f"5 的平方是: {square(5)}")
print(f"10 的平方是: {square(10)}")

# 创建一个专门计算立方的函数
cube = partial(power, exponent=3)
print(f"5 的立方是: {cube(5)}")

# 另一个例子:日志记录
def log_message(level, message):
    """记录带有指定级别的消息。"""
    print(f"[{level.upper()}]: {message}")

# 创建特定级别的日志函数
log_info = partial(log_message, level="info")
log_warning = partial(log_message, level="warning")
log_error = partial(log_message, level="error")

log_info("用户登录成功")
log_warning("磁盘空间不足")
log_error("数据库连接失败")

# 结合高阶函数使用
numbers = [1, 2, 3, 4, 5]
# 过滤出大于等于3的数字
is_greater_than_or_equal_to_3 = partial(lambda x, threshold: x >= threshold, threshold=3)
filtered_numbers = list(filter(is_greater_than_or_equal_to_3, numbers))
print(f"大于等于3的数字: {filtered_numbers}")

wraps
在自定义装饰器中的不可或缺性

如果你写过 Python 装饰器,并且在调试或者查看被装饰函数的信息时遇到过困惑,那么

functools.wraps
就是你的救星。装饰器的本质是一个函数,它接受一个函数作为参数,并返回一个新的函数。问题在于,这个“新函数”会继承装饰器内部的某些属性,而不是被装饰函数本身的属性。

这意味着什么呢?当你调试一个被装饰的函数时,它的

__name__
属性可能不再是原始函数名,而是装饰器内部包装函数的名称;它的
__doc__
字符串也会丢失,变得难以理解;甚至文件路径和行号也可能指向错误的位置。这对于代码的调试、文档生成以及依赖函数元数据的工具(如测试框架、RPC 框架)来说,都是一场灾难。

functools.wraps
就是为了解决这个问题而存在的。它也是一个装饰器,你需要用它来装饰你自定义装饰器内部的“包装函数”。
wraps
会自动将被装饰函数的
__name__
,
__doc__
,
__module__
,
__annotations__
等重要属性复制到包装函数上。

我记得有一次调试一个复杂的系统,好几个装饰器套在一起,结果断点打进去一看,函数名和文档字符串全乱了,简直是噩梦。后来才发现

functools.wraps
的重要性,它就像是给被装饰的函数贴上了一张身份证,让你随时都能认出它本来的面目。这是编写任何通用、可复用装饰器的最佳实践。

from functools import wraps

# 错误的装饰器实现,会丢失被装饰函数的元数据
def my_bad_decorator(func):
    def wrapper(*args, **kwargs):
        print("Something is happening before the function is called.")
        result = func(*args, **kwargs)
        print("Something is happening after the function is called.")
        return result
    return wrapper

@my_bad_decorator
def say_hello_bad(name):
    """一个简单的问候函数."""
    return f"Hello, {name}!"

print("--- 错误装饰器示例 ---")
print(f"函数名: {say_hello_bad.__name__}") # 打印 'wrapper'
print(f"文档字符串: {say_hello_bad.__doc__}") # 打印 None
say_hello_bad("Alice")

# 正确的装饰器实现,使用 @wraps 保持元数据
def my_good_decorator(func):
    @wraps(func) # 关键在这里!
    def wrapper(*args, **kwargs):
        print("Something good is happening before the function is called.")
        result = func(*args, **kwargs)
        print("Something good is happening after the function is called.")
        return result
    return wrapper

@my_good_decorator
def say_hello_good(name):
    """一个更好的问候函数."""
    return f"Hello, {name}!"

print("\n--- 正确装饰器示例 ---")
print(f"函数名: {say_hello_good.__name__}") # 打印 'say_hello_good'
print(f"文档字符串: {say_hello_good.__doc__}") # 打印 '一个更好的问候函数.'
say_hello_good("Bob")

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

753

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

636

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

758

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

618

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1262

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

547

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

577

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

707

2023.08.11

Golang gRPC 服务开发与Protobuf实战
Golang gRPC 服务开发与Protobuf实战

本专题系统讲解 Golang 在 gRPC 服务开发中的完整实践,涵盖 Protobuf 定义与代码生成、gRPC 服务端与客户端实现、流式 RPC(Unary/Server/Client/Bidirectional)、错误处理、拦截器、中间件以及与 HTTP/REST 的对接方案。通过实际案例,帮助学习者掌握 使用 Go 构建高性能、强类型、可扩展的 RPC 服务体系,适用于微服务与内部系统通信场景。

0

2026.01.15

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Django 教程
Django 教程

共28课时 | 3.1万人学习

Excel 教程
Excel 教程

共162课时 | 11.8万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号