傳遞函數函數以獲取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"))