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

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

作用：当我们想给一个函数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秒多

func time is 8.328s

持续更新...