python - Python timeit 设置中的局部变量

Question

<分区>

Answer 1

I would like to use timeit.timeit with variables that are within the scope it is in.

TLDR:

使用 lambda 闭包(之所以这样称呼是因为它关闭了函数中的变量):

def func():
    s1 = 'abc'
    s2 = 'abc'
    return timeit.timeit(lambda: s1 == s2)

我认为这正是您的要求。

>>> func()
0.12512516975402832

解释

所以在全局范围内，你想使用全局变量，而局部范围，本地变量？在全局范围内，locals() 返回与 globals() 相同的结果，因此您 ', '.join(locals())并将其粘贴在 'from __main__ import ' 或 globals() 的末尾，因为它们在全局范围内是等效的:

>>> s1 = 'abc'
>>> s2 = 'abc'
>>> timeit.timeit('s1==s2', 'from __main__ import ' + ', '.join(globals()))
0.14271061390928885

您也可以使用函数和 globals() 来做到这一点，但您不能使用 locals():

s1 = 'abc'
s2 = 'abc'
def func():
    return timeit.timeit('s1==s2', 'from __main__ import ' + ', '.join(globals()))

和

>>> func()
0.14236921612231157

但以下内容不起作用，因为您必须从 import 语句访问隐藏在函数局部范围内的变量:

def func():
    s1 = 'abc'
    s2 = 'abc'
    return timeit.timeit('s1==s2', 'from __main__ import ' + ', '.join(locals()))

但是因为您可以简单地将函数传递给 timeit，所以您可以做的是:

def func(s1='abc', s2='abc'):
    s1 == s2

和

>>> timeit.timeit(func)
0.14399981498718262

这也意味着，在您的函数中，您可以为 timeit 提供一个 lambda 闭包:

def func():
    s1 = 'abc'
    s2 = 'abc'
    return timeit.timeit(lambda: s1 == s2)

或全功能定义:

def func():
    s1 = 'abc'
    s2 = 'abc'
    def closure():
        return s1 == s2
    return timeit.timeit(closure)

我认为这正是您的要求。

>>> func()
0.12512516975402832

When they're both not in the main program

如果您想从 __main__ 以外的其他模块加入全局变量而不是设置，请使用:

'from ' + __name__ + ' import ' + ', '.join(globals())