, wie viel Zeit eine Funktion nehmen

Question

i in Python ausführen, um eine Funktion definiert, sagen

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

die einige Operationen ausgeführt und das Ergebnis in der Datenbank

und ich habe eine Tabelle in Pandas Datenrahmen die hat ID-Spalte drin, und etwa 1000 Zeilen mit eindeutigen IDs

ich werde diese Funktion mit .apply Funktion aufrufen.

df['id'].apply(func1) 

, die eine ID zu einer Zeit nehmen und tun Operationen und speichern sie in der Datenbank, und gleiche wird für alle ID in meinem Datenrahmen geschehen.

Gibt es eine Möglichkeit zu berechnen, wie viel Gesamtzeit meine Funktion dauerte, um für alle IDs in df auszuführen?

ich dachte, vielleicht verwenden time Paket und tun smthing damit. aber ich bin ein Anfänger.

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

Gibt es eine Möglichkeit, dies zu tun?

Answer 1

Sie können auch den Kontextmanager verwenden, um es zu überprüfen. Es könnte etwas Zeit sparen, wenn Sie es wiederverwenden möchten.

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