Question

我定义了一个函数，比如说

def func1(id):
    ....
    logic
    dataframe2.to_sql(query)
    ....

运行一些操作并将结果存储在数据库中

我在pandas dataframe中有一个表，其中包含id列，大约有1000行具有不同的id

我将使用.apply函数调用此函数。

df['id'].apply(func1)

一次只占用一个id并执行操作并将其存储在数据库中，对于我的数据帧中的所有id也会发生同样的情况。

有没有办法计算我的功能运行总时间对于df中的所有id？

我想也许可以使用time包并用它做点什么。但我是初学者。

import time
start_time = time.time()
main()
print("--- %s seconds ---" % (time.time() - start_time))

有没有办法做到这一点？

Answer 1

您也可以使用上下文管理器进行检查。如果你想重复使用它可以节省一些时间。

>>> class TimeMyFunc(object):
...      def __enter__(self):
...          self.begin = time.time()
...      def __exit__(self,*args):
...          elapsed = time.time() - self.begin
...          print elapsed
...          
>>> with TimeMyFunc():
...     time.sleep(1)
...     
1.0

函数在python中运行需要多长时间

1 个答案: