Python：如何使用timeit？

Question

我尝试编写一个装饰器来计算 Python 中另一个函数的运行时间。

def timer(func):
        def smallfunctimer(*args,**kwargs):
            res=func(*args,**kwargs)
            tm=timeit.Timer(lambda:res)
            print("Function time: ", tm)
        return smallfunctimer

@timer
    def deposit(self,amount):
        self.balance+=amount
        self.queue.append(BankTransaction(amount))

但是当我调用它时

ba=BankAccount(1,100)
ba.deposit(10000000)

i get this:

Function time:  <timeit.Timer object at 0x0281D370>

如何获取运行时间（以秒为单位）？

- Igor Achmiz

Answer 1

您创建了一个timeit.Timer()实例，您需要调用该实例的方法才能实际运行代码。请注意保留HTML标签。

print("Function time: ", tm.timeit(1000))

这个函数运行1000次所需的时间将会被返回。

- Martijn Pieters