使用Python的timeit来定义"全局名称'foo'"

Question

我试图找出执行Python语句需要多长时间,所以我在线查看并发现标准库提供了一个名为timeit的模块,其目的是做到这一点:

import timeit

def foo():
    # ... contains code I want to time ...

def dotime():
    t = timeit.Timer("foo()")
    time = t.timeit(1)
    print "took %fs\n" % (time,)

dotime()

但是,这会产生错误:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 3, in dotime
  File "/usr/local/lib/python2.6/timeit.py", line 193, in timeit
    timing = self.inner(it, self.timer)
  File "<timeit-src>", line 6, in inner
NameError: global name 'foo' is not defined

我还是Python新手,我不完全理解它所有的范围问题,但我不知道为什么这个片段不起作用.有什么想法吗？

Answer 1

改变这一行:

t = timeit.Timer("foo()")

对此:

t = timeit.Timer("foo()", "from __main__ import foo")

查看最底部提供的链接.

要使timeit模块能够访问您定义的函数,可以传递包含import语句的setup参数:

我只是在我的机器上测试它,它适用于变化.

Answer 2

你可以试试这个黑客:

import timeit

def foo():
    print 'bar'

def dotime():
    t = timeit.Timer("foo()")
    time = t.timeit(1)
    print "took %fs\n" % (time,)

import __builtin__
__builtin__.__dict__.update(locals())

dotime()

Answer 3

使用Python 3,您可以使用 globals=globals()

t = timeit.Timer("foo()", globals=globals())

从文档:

另一种选择是传递globals()给globals参数,这将导致代码在当前的全局命名空间中执行.这比单独指定导入更方便

Answer 4

t = timeit.Timer("foo()", "from __main__ import foo")

由于timeit没有你的东西在范围内.