使用 Python 的 timeit 测量代码执行时间
在 Python 中,量化代码段的执行时间对于性能测试至关重要。本文探讨了如何利用 Python 的 timeit 模块来完成此任务。
示例用例
考虑以下在数据库上执行多个查询的 Python 脚本:
<code class="python">conn = ibm_db.pconnect("dsn=myDB", "usrname", "secretPWD") for r in range(5): print "Run %s\n" % r query_stmt = ibm_db.prepare(conn, update) ibm_db.execute(query_stmt) ibm_db.close(conn)</code>
为了测量查询的执行时间,我们可以使用 Python 的 timeit 模块。
使用 timeit
第 1 步:导入timeit 模块
<code class="python">import timeit</code>
第 2 步:定义代码到时间
创建一个字符串或函数,封装需要测量执行时间的代码:
<code class="python">setup_code = """ import ibm_db conn = ibm_db.pconnect("dsn=myDB","usrname","secretPWD") query_stmt = ibm_db.prepare(conn, update) """ code_to_time = """ ibm_db.execute(query_stmt) """</code>
第 3 步:设置配置参数
指定 timeit 函数的重复次数和迭代次数:
<code class="python">repetitions = 5 iterations = 100</code>
第 4 步:测量执行时间
<code class="python">timeit_result = timeit.timeit(code_to_time, setup=setup_code, number=iterations, globals=globals())</code>
第 5 步:输出结果
<code class="python">print("Execution time:", timeit_result)</code>
其他注意事项
通过执行以下步骤,您可以准确地为代码段计时在您的 Python 脚本中并深入了解其性能特征。
以上是如何使用Python的“timeit”模块来测量代码段的执行时间?的详细内容。更多信息请关注PHP中文网其他相关文章!