一尘不染

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

python

我试图找出执行一条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还是很陌生,我还不完全了解它的所有范围界定问题,但是我不知道为什么此代码片段不起作用。有什么想法吗?


阅读 139

收藏
2020-12-20

共1个答案

一尘不染

更改此行:

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

对此:

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

检查一下您在最底部提供的链接。

要使timeit模块可以访问您定义的功能,可以传递一个包含导入语句的设置参数:

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

2020-12-20