Python 函数返回值只取其中一个的实现与装饰器处理

唐可

发布时间：2026-10-15| 4人浏览过

原创

本文介绍 Python 中通过解包赋值只获取函数部分返回值的方法，并深入探讨装饰器在处理有返回值函数时的实现细节。通过代码示例展示了如何利用 functools.wraps 保留元信息，并对比了 for 循环与 map 匿名函数的运算速度，为开发者提供实用的性能优化参考。

首先通过代码示例来直观展示如何从函数返回值中提取特定值。


def test():
    return 1,2
a, b = test()
1 2
a, _ = test()
1   
 
# 说明 a, _ = test() 中 返回的是一个int类型 后续可直接运算
# a, b = test()  返回值是一个元组, 直接运算报错

补充内容：探讨 Python 装饰器在被装饰函数存在返回值情况下的处理方式。

具体题目要求如下：

利用装饰器实现一个函数计时器，用于比较 for 循环与 map 匿名函数的运算速度。

遇到的核心问题：

当被装饰的函数本身带有返回值时，应该如何正确处理？

具体的程序实现代码如下：


import time
import random
import string
import functools
#问题：被装饰的函数有返回值
li = [random.choice(string.ascii_letters) for i in range(100)]
def timeit(fun):
    @functools.wraps(fun)  #为了保留被装饰函数的函数名和帮助文档信息
    def wrapper(*args,**kwargs):
        """这是一个wrapper函数"""
        start_time = time.time()
        res = fun(*args,**kwargs)
        end_time = time.time()
        print('运行时间为:%.6f' %(end_time - start_time))
        return res
    return wrapper
@timeit
def fun_list(n):
    """这是fun_list函数"""
    return [i * 2 for i in range(n)]
@timeit
def fun_map(n):
    """这是fun_map函数"""
    return list(map(lambda x:x*2,range(n)))
print(fun_list(5000))
print(fun_map(5000))