使用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模块可以访问您定义的功能，可以传递一个包含导入语句的设置参数：

我刚刚在我的机器上对其进行了测试，并且可以进行更改。