如何通過Python在其他函數中用參數調用一個函數？

Question

我知道標題有點混亂。

因此，讓我藉此代碼爲例：

import timeit 

def test1(x,y): 
    return x*y 

def time_cost(func): 
    start = timeit.default_timer() 
    func() 
    stop = timeit.default_timer() 

    print(stop - start) 

time_cost(test1) 

我想給test1的兩大參數，x和y，以time_cost功能。

但我不知道該怎麼做。

我曾嘗試在此之前：

import timeit 

def test1(x,y): 
    return x*y 

def time_cost(func): 
    start = timeit.default_timer() 
    func 
    stop = timeit.default_timer() 

    print(stop - start) 

time_cost(test1(1,2)) 

它可以運行，但結果是如此怪異，我認爲這是錯誤的。

那麼如何做到這一點？謝謝你們。

Answer 1

返回一個你想要的功能。

def time_cost(func): 
    def run(*args, **kwargs): 
     start = timeit.default_timer() 
     func(*args, **kwargs) 
     stop = timeit.default_timer() 

     print(stop - start) 
    return run 

然後調用它。

time_cost(test1)(1, 2) 

Answer 2

試試這個。

from functools import wraps 
import timeit 

def time_cost(func): 
    @wraps(func) 
    def wrapper(*args, **kwargs): 
     start = timeit.default_timer() 
     result = func(*args, **kwargs) 
     stop = timeit.default_timer() 
     print(stop - start) 
     return result 
    return wrapper 

test1 = time_cost(test1) 
test1(1,2) 

這種函數被稱爲裝飾器。也稱爲包裝，或封閉。

您還可以使用修飾語法使用：

@time_cost 
def test1(x,y): 
    return x*y 

test1(1,2) 

這並不與上面相同的代碼。