传递函数函数以获取NameError的timeit

Question

我想用timeit计算三个类似的函数。 我已经编写了这段代码，但是当我将一个函数传递给测试函数时，我不明白发生了什么。

def f0(x,  y,  z):
#some code here

def f1(x,  y,  z):
#a slighty similar function


def f2(x, y, z):
#still another similar function



def test(name):
    x=100
    y=100
    z=100
    res=name(x,y,z)

if __name__ == '__main__':
    import timeit
    print(timeit.timeit("test(f0)", setup="from __main__ import test"))

我得到的错误是：

NameError: global name 'f0' is not defined

Answer 1

您需要导入测试中提到的所有全局名称：

print(timeit.timeit("test(f0)", setup="from __main__ import test, f0"))

线路test(f0)需要查找f0 ，而不仅仅是test() 。

Answer 2

你也必须导入f0 ，就像这样

print(timeit.timeit("test(f0)", setup="from __main__ import test, f0"))