Timeit， NameError： 未定义全局名称。但我没有使用全局变量

Question

我想测量以下代码的执行速度：

def pe1():
    l = []
    for i in range(1000):
        if i%3 == 0 or i%5 == 0:
            l.append(i)
    print sum(l)

我将此代码存储在 pe1m.py .现在我想用python解释器测试文件的速度。我做了：

import timeit
import pe1m
t = timeit.Timer(stmt = 'pe1m.pe1()')
t.timeit()

但我得到：

File "<stdin>", line 1, in <module>
  File "/usr/lib/python2.7/timeit.py", line 195, in timeit
    timing = self.inner(it, self.timer)
  File "<timeit-src>", line 6, in inner
NameError: global name 'pe1m' is not defined

但我没有任何全局变量。

Answer 1

试试这个：

t = timeit.Timer(stmt='pe1()', setup='from pe1m import pe1')

timeit.Timer对象不知道要在其中调用它的命名空间，因此它无法访问导入的pe1m模块。

setup参数是在定时语句的上下文中执行的语句，它们共享相同的命名空间，因此无论您在那里定义什么，都可以在stmt中访问。

Answer 2

你也可以试试这个

>>>def pe1():
>>>    l = []
>>>    for i in range(1000):
>>>        if i%3 == 0 or i%5 == 0:
>>>            l.append(i)
>>>    print(sum(l))
>>>
>>>from  timeit import time it
>>>timeit('pe1()',setup="from __main__ import pe1",number=100) # Run 100 times
>>>timeit('pe1()',setup="from __main__ import pe1")  # run default times of 1000000