Python装饰器计算函数执行时间

时过境迁_o

已于 2022-04-03 22:47:36 修改

阅读量3.8k

点赞数 2

CC 4.0 BY-SA版权

文章标签：冒泡排序算法 python list 开发语言

于 2022-04-03 22:44:26 首次发布

本文链接：https://blog.csdn.net/wangzhiping321/article/details/123946083

本文介绍了Python装饰器的基本概念，它允许在不修改原有函数代码的情况下添加额外功能。通过一个示例展示了如何使用装饰器来计算冒泡排序函数的执行时间。在示例中，创建了一个名为`wrapper`的装饰器，用于记录并打印函数的运行时间。随后，定义了一个冒泡排序函数，并使用`@wrapper`装饰器进行装饰，以统计其在处理随机生成的8000个整数列表时的耗时，结果显示冒泡排序执行耗时约8.328秒。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

装饰器本质就是一个函数：在函数内又定义了函数，内部函数引用了外部函数的变量，然后外部函数将内部函数作为值给返回；

作用：当我们想给一个函数A增加新功能，但我们又不想改变函数A的时候，装饰器就派上用场了。

定义一个装饰器函数（默认用*args, **kwargs表示不定长参数）

import time

def wrapper(func):
    def inner(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        end_time = time.time()
        result = end_time - start_time
        print('func time is %.3fs' % result)
        return res
    return inner

定义一个冒泡排序的函数：

此时我们想统计冒泡排序函数的执行时长，可以使用Python语法糖的格式，在函数的上面@+装饰器函数名称

import random
# 在0-20000数字之间随机取8000个整数组成一个列表，并对其使用冒泡排序
n = [i for i in range(20000)]
arr = random.sample(n, 8000)

@wrapper
def bubble_sort(arr):
    for i in range(len(arr) -1, 0, -1):
        for j in range(i):
            if arr[j] > arr[j + 1]:
                arr[j], arr[j + 1] = arr[j + 1], arr[j]
    return arr

bubble_sort(arr)

执行结果：

bubble_sort 执行耗时8秒多