import time
def time_use(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
测试代码:
import random
# 在0-20000数字之间随机取8000个整数组成一个列表,并对其使用冒泡排序
n = [i for i in range(20000)]
arr = random.sample(n, 8000)
@time_use
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)