Python timeit模块的使用实践

  insert_time_repeat = timeit.repeat(stmt='list(insert_list.insert(0, i) for i in init_list)',
                    setup='insert_list=list();init_list=range(10)',
                    repeat=5,
                    number=10000)
  print('insert_time_repeat: ', insert_time_repeat)
  append_time_repeat = timeit.repeat(stmt='list(append_list.append(i) for i in init_list)',
                    setup='append_list=list();init_list=range(10)',
                    repeat=5,
                    number=10000)
  print('append_time_repeat: ', append_time_repeat)

运行结果如下：

('insert_time_repeat: ', [2.591015, 2.5814996999999997, 2.5547322, 2.6153070000000005, 2.5496864000000006])
('append_time_repeat: ', [0.0181692999999985, 0.01746889999999901, 0.018901899999999472, 0.018737400000000903, 0.018211900000000725])

三、使用 Timer 类测试运行时间

1. Timer 类介绍

上面使用了 timeit() 函数和 repeat() 函数，其实在 timeit 模块中，这两个函数都是对 Timer 类做了进一步的封装，实际调用的还是 Timer 类中的方法。

在 Timer 类中，实现了两个方法，timeit() 方法和 repeat() 方法，上面两个函数调用的就是这两个方法。

在使用  from timeit import … 时，只能导入 Timer 类，所以可以直接使用 Timer 类来测试，可以自己去调用方法，使用起来更灵活。

2. 测试列表头部添加

先实例化一个 Timer 类的对象，实例化时传入 stmt 和 setup 参数(参数的含义与上面一致)，timer 参数保持默认，然后通过实例对象调用对应的 timeit() 方法或 repeat() 方法，在 timeit() 方法中传入 number，在repeat() 方法中传入 number 和 repeat 。

使用 timeit() 方法和 repeat() 方法测试从头部添加数据的运行时间。

 timer_insert = timeit.Timer(stmt='insert_time_test()', setup='from __main__ import insert_time_test')
  insert_time_timeit = timer_insert.timeit(number=1000000)
  print('insert_time_timeit: ', insert_time_timeit)
  insert_time_repeat = timer_insert.repeat(number=1000000)
  print('insert_time_repeat: ', insert_time_repeat)

运行结果如下：

('insert_time_timeit: ', 2.7732486)
('insert_time_repeat: ', [2.7367806999999997, 2.707402600000001, 2.7288245999999994])

3. 测试列表尾部添加

使用 timeit() 方法和 repeat() 方法测试从尾部添加数据的运行时间。

  timer_append = timeit.Timer(stmt='append_time_test()', setup='from __main__ import append_time_test')
  append_time_timeit = timer_append.timeit(number=1000000)
  print('append_time_timeit: ', append_time_timeit)
 3/4   首页 上一页 1 2 3 4 下一页 尾页