Python 时间问题

Question

我正在尝试使用 timeit 模块，但我不知道如何使用。 我有一个 main:

from Foo import Foo
if __name__ == '__main__':
...
   foo = Foo(arg1, arg2) 
   t = Timer("foo.runAlgorithm()")
   print t.timeit(2)

并且我的 Foo 类有一个名为 runAlgorithm() 的方法，

错误是这样的：

NameError：全局名称“foo”未定义

我做错了什么？ 我可以从类方法中抽出时间吗？

Answer 1

而不是使用必要的 setup 参数设置 timeit 环境，您可以简单地传递方法（或任何可调用的内容）：

t = Timer(foo.runAlgorithm)

来自文档：

版本 2.6 中的更改：stmt 和 setup 参数现在也可以采用无需参数即可调用的对象。

如果需要传递一些参数，可以使用 functools 进行函数柯里化。部分，例如：

class C:
    def printargs(self, a, b):
        print a, b

from functools import partial
foo = C()
t = Timer(partial(foo.printargs, 1, 2))

Answer 2

正如文档示例所示，您需要传递setup语句：

t = Timer("foo.runAlgorithm()", 'from __main__ import foo')