从 timeit 访问外部变量

Question

我想为一堆语句计时。例如：

s = """\
for i in range (0,3):
    for j in range (0,4):
        if alpha_beta[i,j] == 0 and vect2[i] != 0.0 and alpha[j] == 1 :
            print(i,j)
            alpha_beta[i,j] = vect2[i]
"""
timeit.timeit(stmt=s, number=100)

但它抛出错误：name 'alpha_beta' is not defined。

变量 alpha_beta、vect2 和 alpha 定义较早。我不能将它们包括在计时器中，因为那也会包括它们的初始化开销。

我是 Python 的新手，非常感谢修改我的代码的答案。

Answer 1

您可以将它们导入 setup:

timeit.timeit(stmt=s, setup='from __main__ import alpha_beta, vect2, alpha', number=100)

从 Python 3.5 开始 timeit.timeit 也可以在提供的 globals 命名空间中执行代码。将其设置为 globals() 将使 stmt 能够访问模块命名空间中的名称：

timeit.timeit(stmt=s, number=100, globals=globals())

Answer 2

一种解决方案是提供函数而不是字符串。这样您的代码就可以轻松地在其创建的上下文中访问全局变量。

也就是说，您的代码将变为：

def f():
    for i in range (0,3):
        for j in range (0,4):
            if alpha_beta[i,j] == 0 and vect2[i] != 0.0 and alpha[j] == 1 :
                print(i,j)
                alpha_beta[i,j] = vect2[i]   

timeit.timeit(stmt=f, number=100)

从 timeit 访问外部变量

Accessing outside variables from timeit

python

timeit