如何在每个测试都需要随机设置的情况下使用 timeit

Posted 2023-03-06

【中文标题】如何在每个测试都需要随机设置的情况下使用 timeit

【英文标题】：How to use timeit where each test requires random setup

【发布时间】：2013-10-26 11:34:20

【问题描述】：

我有一个函数f(x)，它将一个列表x作为输入，其中包含0到1之间的100个随机浮点数。不同的列表将导致f的不同运行时间。

我想知道f 在大量不同的随机列表中平均运行多长时间。最好的方法是什么？我应该使用timeit吗？如果可以，有没有一种方法可以做到这一点，而不包括在每次试验中生成每个随机列表所需的时间？

如果没有timeit（伪代码），我会这样做：

for i = 1 to 10000:
    x = random list
    start = current time
    f(x)
    end = current time
    results.append(end - start)
return mean(results)

【问题讨论】：

***.com/questions/2245161/…

@lnwvr 哈哈，我从没见过。如果我这样做了，我只是复制它，不需要自己制作装饰器！ :P

【参考方案1】：

你可以做一个计时器装饰器：

下面是一些示例代码：

from time import time


class Timer(object):
    def __init__(self, func):
        """
        Decorator that times a function
        @param func: Function being decorated
        @type func: callable
        """
        self.func = func

    def __call__(self, *args, **kwargs):
        start = time()
        self.func(*args, **kwargs)
        end = time()
        return end - start


@Timer
def cheese():
    for var in xrange(9999999):
        continue

for var in xrange(100):
    print cheese()

工作 example，循环更少。

【参考方案2】：

import timeit, random

def summer(myList):
    result = 0
    for num in myList:
        result += num
    return result

for i in range(10):
    x = [random.randint(0, 100) for i in range(100000)]
    print timeit.timeit("summer(x)", setup="from __main__ import x, summer", number = 100)

您可以使用from __main__ import x 导入变量

【讨论】：

在循环中多次调用timeit 会破坏使用它的目的吗？

@Andrew 你想每次都用随机输入测试函数，对吧？这就是为什么它在一个循环中，每次都随机输入。

@Andrew 相同，但使用timeit 的方式。

timeit 没有我想要的那么灵活。尽管如此，这是一种实现您想要的方式的方法。

【参考方案3】：

我认为这可以解决问题。它将每次重复执行一次setup，然后执行一次stmtnumber=1。但是，我认为这并不比您发布的简单循环好多少。

import timeit

stmt = '[x*x*x for x in xrange(n)]'  # just an example 
setup = 'import random; n = random.randint(10, 100)'
r = 10000
times = timeit.repeat(stmt, setup, repeat=r, number=1)

print min(times), max(times), sum(times)/r

还有一个“单元格模式”可以与timeit in the IPython shell 一起使用，但它只返回禁食时间，并且没有简单的方法可以改变它（？）。

import random

%%timeit -r 10000 -n 1 n = random.randint(10,100)
var = [x*x*x for x in xrange(n)]