python - 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)