获取“全局名称‘foo’” 未定义” 使用Python的timeit

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 语句的设置参数：

我刚刚在我的机器上测试了它，并且它可以处理更改。

Answer 2

globals=globals()

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

对于 Python 3，您可以使用

另一个选项是将 globals() 传递给 globals 参数，该参数
将导致代码在您当前的全局中执行
命名空间。 这比单独指定更方便
进口

Answer 3

你可以尝试这个技巧：

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 4

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

因为 timeit 的范围内没有你的东西。

Answer 5

添加到您的设置中“import thisfile;”

然后当您调用设置函数 myfunc() 时使用“thisfile.myfunc()”

例如“thisfile.py”

def myfunc():

 return 5

def testable(par):

 pass



t=timeit.timeit(stmt="testable(v)",setup="import thisfile; v=thisfile.myfunc();").repeat(10)

print( t )