Python Timeit 和“全局名称...未定义”

Question

我对代码优化的 timit 函数有疑问。例如，我在一个文件中编写带有参数的函数，我们将其称为包含 : 的 myfunctions.py ，

def func1(X):
    Y = X+1
    return Y

并在第二个文件 test.py 中测试此函数，我在其中调用用于测试代码性能的计时器函数（在显然更复杂的问题中！）包含：

import myfunctions
X0 = 1
t = Timer("Y0 = myfunctions.func1(X0)")
print Y0
print t.timeit()

未计算 Y0，即使我注释 print Y0 行错误 global名称“myfunctions”未定义发生了。

如果我现在使用命令指定设置

t = Timer("Y0 = myfunctions.func1(X0)","import myfunctions")

，则会发生错误全局名称“X0”未定义。

有人知道如何解决这个问题吗？非常感谢。

Answer 1

您需要setup参数。尝试：

Timer("Y0 = myfunctions.func1(X0)", setup="import myfunctions; X0 = 1")

Answer 2

Y0 未定义的原因是您已在字符串中定义它，但在执行开始的解析时，尚未对字符串进行求值以使变量生效。因此，在脚本顶部的某个位置放置一个 Y0 = 0 来预先定义它。

所有外部函数和变量都必须使用其 setup 参数提供给 Timer。因此，您需要 "import myfunctions; X0 = 1" 作为设置参数。

这将起作用：

from timeit import Timer
import myfunctions
X0 = 1
Y0 = 0     #Have Y0 defined
t = Timer("Y0 = myfunctions.func1(X0)", "import myfunctions; X0 = %i" % (X0,))
print t.timeit()
print Y0

看看我如何使用 "X0 = %i" % (X0,) 传入外部 X0 变量的实际值。

您可能想知道的另一件事是，如果您的主文件中有任何想要在 timeit 中使用的函数，您可以通过传递 timeit 来识别它们>from __main__ import * 作为第二个参数。

---

如果您希望 timeit 能够修改变量，那么您不应该向它们传递字符串。您可以更方便地将可调用对象传递给它。您应该传递一个可更改所需变量的可调用函数。那么您就不需要设置了。看：

from timeit import Timer
import myfunctions

def measure_me():
    global Y0    #Make measure_me able to modify Y0
    Y0 = myfunctions.func1(X0)

X0 = 1
Y0 = 0     #Have Y0 defined
t = Timer(measure_me)
print t.timeit()
print Y0

如您所见，我将 print Y0 after print t.timeit() 放在执行之前，因为在执行之前您无法更改其值！