定义一个装饰器来自动测量时间

最新推荐文章于 2022-03-04 16:42:13 发布

原创最新推荐文章于 2022-03-04 16:42:13 发布 · 149 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一个实用的Python技巧，通过定义一个装饰器来自动测量函数的执行时间，帮助开发者更方便地进行性能分析和优化。

定义一个修饰器来自动测量时间

from functools import wraps
def timefn(fn):
	@wraps(fn)
 	def measure_time(*args, **kwargs):
		t1 = time.time() 
 		result = fn(*args, **kwargs)
		t2 = time.time()
 		print ("@timefn:" + fn.__name__ + " took " + str(t2 - t1) + " seconds")
		return result
 	return measure_time


@timefn
def calculate_z_serial_purepython(maxiter, zs, cs):
 ...

关注博主即可阅读全文