python - 将 lambda 函数作为变量传递给 timeit 中的另一个函数

Question

我有一个名为 func 的 Python 函数，我正在尝试使用 timeit 计时。

func 有一个参数，f，它也是一个函数。在我正在编写的程序中，它是一个 lambda 函数，但它也可以是一个常规定义的函数。重要的是此函数的源代码对 func 不可用，所有可用的是对它的引用。

我正在尝试做的事情:

def testTimingOfFunc(f):
    time = timeit.timeit("package.func(f)", "from src import package")

我收到一条错误消息，提示未定义 f。

其他一切正常。当我打电话时:

testTimingOfFunc('lambda x:x**2')

并将函数更改为:

def testTimingOfFunc(f):
    time = timeit.timeit("package.func(" + str(f) + ")",
                         "from src import package")

一切正常，所以我知道关于 timeit 函数的一切都很好。

testTimingOfFunc 不是这样工作的；传入的是函数引用，而不是字符串。

关于如何将 f 作为参数传递给 timeit 中的 func 有什么想法吗？

Answer 1

可以使用 python 可调用值调用 timeit:只需将您的方法更改为:

from src import package
import functools

def testTimingOfFunc(f):
    time = timeit.timeit(functools.partial(package.func, f))