python中获取执行时间函数

Question

我创建这个工作函数是为了获取另一个函数的时间：

def get_execution_time(function, args, numberOfExecTime=1):
    """Return the execution time of a function in seconds.

    """

    return round(Timer(partial(function, args)).timeit(numberOfExecTime), 5)

顺便说一句，我有一个问题：我无法向要计时的函数提供多个输入（参数）。 我怎样才能做到这一点？部分是正确的工具吗？

我尝试了装饰器，但无法存储进行一些统计所需的时间。

Answer 1

如果您可以牺牲 numberOfExecTime 参数具有默认值的能力，您可以这样做：

from timeit import Timer
from functools import partial

def get_execution_time(function, numberOfExecTime, *args, **kwargs):
    """Return the execution time of a function in seconds."""
    return round(Timer(partial(function, *args, **kwargs))
                 .timeit(numberOfExecTime), 5)

def foo(a, b, c = 12):
    print a, b, c

get_execution_time(foo, 1, 3, 4, c = 14)

或者您可以这样做，但仍然具有 numberOfExecTime 的默认值：

from timeit import Timer
from functools import partial

def get_execution_time(function, *args, **kwargs):
    """Return the execution time of a function in seconds."""
    numberOfExecTime = kwargs.pop('numberOfExecTime', 1)
    return round(Timer(partial(function, *args, **kwargs))
                 .timeit(numberOfExecTime), 5)

def foo(a, b, c = 1):
    print a, b, c

get_execution_time(foo, 1, 2, c = 2)
# => 1 2 2

get_execution_time(foo, 4, 5, c = 3, numberOfExecTime = 2)
# => 4 5 3
# => 4 5 3

Answer 2

您可以传递多个参数：args 应该是包含要传递的参数的元组。

现在，您可以传递 *args 而不是 args。

（如果您还需要将 kwargs 传递给您的方法，则在 get_execution_time 中添加另一个参数 kwargs 并将 **kwargs 传递给您的 函数)

def get_execution_time(function, args=(), kwargs ={}, numberOfExecTime=1):
    return round(Timer(partial(function, *args, **kwargs)).timeit(numberOfExecTime), 5)

Answer 3

我会使用计时装饰器。

import time

def timeit(f):

    def timed(*args, **kw):

        ts = time.time()
        result = f(*args, **kw)
        te = time.time()

        print 'func:%r args:[%r, %r] took: %2.4f sec' % \
          (f.__name__, args, kw, te-ts)
        return result

    return timed

使用装饰器很容易或者使用注释。

@timeit
def compute_magic(n):
     #function definition
     #....

或者为您想要计时的函数重新设置别名。

compute_magic = timeit(compute_magic)

我的博客文章在这里有更多信息。 http://blog.mattalcock.com/2013/2/24/计时-python-代码/

Answer 4

如果 args 是参数列表，您可以使用 *args 扩展它：

return round(Timer(partial(function, args)).timeit(numberOfExecTime), 5)