如何使用Python的timeit计时代码段以测试性能？

Question

我有一个Python脚本，该脚本可以正常工作，但是我需要编写执行时间。我已经用谷歌搜索了，timeit但是我似乎无法使它正常工作。

我的Python脚本如下所示：

import sys
import getopt
import timeit
import random
import os
import re
import ibm_db
import time
from string import maketrans
myfile = open("results_update.txt", "a")

for r in range(100):
    rannumber = random.randint(0, 100)

    update = "update TABLE set val = %i where MyCount >= '2010' and MyCount < '2012' and number = '250'" % rannumber
    #print rannumber

    conn = ibm_db.pconnect("dsn=myDB","usrname","secretPWD")

for r in range(5):
    print "Run %s\n" % r        
    ibm_db.execute(query_stmt)
 query_stmt = ibm_db.prepare(conn, update)

myfile.close()
ibm_db.close(conn)

我需要的是执行查询并将其写入文件所需的时间results_update.txt。目的是使用不同的索引和调整机制来测试数据库的更新语句。

Answer 1

您可以在要计时的块之前或之后使用time.time()或time.clock()。

import time

t0 = time.time()
code_block
t1 = time.time()

total = t1-t0

此方法不完全精确timeit（它不会平均运行几次），但是很简单。

time.time()（在Windows和Linux中）和time.clock()（在Linux中）不够精确，无法实现快速功能（total =
0）。在这种情况下，或者如果要平均几次运行所花费的时间，则必须多次手动调用该函数（就像我在示例代码中已经做过的那样，并且在设置其 number
参数时timeit会自动执行）

import time

def myfast():
   code

n = 10000
t0 = time.time()
for i in range(n): myfast()
t1 = time.time()

total_n = t1-t0

如注释中所述，在Windows中，Coreytime.clock()具有更高的精度（微秒而不是秒），并且优于time.time()。