使用timeit.Timer（）时如何传递函数的参数

Question

这是一个简单程序的概述

# some pre-defined constants
A = 1
B = 2

# function that does something critical
def foo(num1, num2):
    # do something

# main program.... do something to A and B
for i in range(20):
    # do something to A and B
    # and update A and B during each iteration

import timeit
t = timeit.Timer(stmt="foo(num1,num2)")  
print t.timeit(5)

我只是不断获取“未定义全局名称foo” .....有人可以帮我吗？谢谢！

Answer 1

代码段必须是独立的-它们不能进行外部引用。您必须在statement-string或setup-string中定义值：

import timeit

setup = """
A = 1
B = 2

def foo(num1, num2):
    pass

def mainprog():
    global A,B
    for i in range(20):
        # do something to A and B
        foo(A, B)
"""

t = timeit.Timer(stmt="mainprog()" setup=setup)
print(t.timeit(5))

更好的是，重写代码以不使用全局值。