如何通过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) 

这并不与上面相同的代码。