猫になりたい

コンサルのデータ分析屋、計量経済とか機械学習をやっています。pyてょnは3.7を使ってマスコレルウィンストングリーン。

pythonで関数の実行時間を計測する

pyhtonで関数の実行時間を計測したい時今までは
qiita.com
に書いてある方法で実行したい部分を挟んであげていたんですが 毎回3行挿入するのは面倒だし読みにくいので任意の関数の実行時間を計測するモジュールを作成しました。

誰かが似たようなことやってそうだけど調べてはいません。

コードは以下のgithubに上げてあります
github.com
まずtimethis.pyをインポートしたら

@timethis
def counter():
    c = 0
    for i in range(10000000):
        c+=1
    print(c)
    
counter()

とするか

def counter():
    c = 0
    for i in range(10000000):
        c+=1
    print(c)
    
counter = timethis(counter)
counter()

としてあげれば以下のような出力が得られます。

output 出力

--start wrapper --
10000000
elapsed_time:elapsed_time:0.915 [sec] --end--