導引
在計算密集型計算或一些Web應用中,我們常常需要對代碼做性能分析。在Python中,最原始的方法即是使用time
包中的time
函數(該函數以秒為計時單位):
from time import sleep, time
def func1():
sleep(0.001)
def func2():
sleep(2)
begin1 = time()
func1()
end1 = time()
begin2 = time()
func2()
end2 = time()
print("func1 consume: %f, func2 consume:%f, func3 consume: %f"\
% (end1-begin1, end2-begin2, end2-begin1))
控制台輸出如下:
func1 consume: 0.001271, func2 consume:2.000421, func3 consume: 2.001692
pyinstrument基本用法
但是一旦函數多了起來,這種方式顯然過於繁雜。類似C語言中的cProfile,在Python中,也有專門的性能分析工具pyinstrument
(該庫非內置,需要使用conda/pip安裝),我們在復雜的項目中可以使用它來代替簡陋的time.time()
。
首先來看一下基本的使用,它的使用框架如下:
from pyinstrument import Profiler
from time import sleep
def func1():
sleep(0.1)
def func2():
sleep(2)
profiler = Profiler()
profiler.start()
# 這里是你要分析的代碼,我們這里分析func1和func2兩個函數
func1()
func2()
profiler.stop()
profiler.print()
可以看到,該工具也將其成功分析出了個函數的運行時間,並為我們標紅指出了運行2s的func2函數是性能瓶頸:

如果我們進一步調低函數的運行時間:
def func3():
sleep(0.0001)
profiler = Profiler()
profiler.start()
func3()
profiler.stop()
profiler.print()
此時會顯示“No samples were recorded”,如下:
這是因為你的代碼運行時間小於了1ms,如果你仍然想分析這段代碼,你可以選擇將間隔值調到比默認的0.001(1ms)小,比如這樣:
profiler = Profiler(interval=0.0001)
此時你會發現,func3也能被檢測出來了:

此外,如果你要在瀏覽器中查看分析結果,可以使用profiler.open_in_browser()
代替profiler.print()
的控制台打印輸出:

也可以使用profiler.output_html()
將profile以html形式輸出。
分析Flask中的web響應性能
我們也可以對Flask應用進行性能分析,具體的用法如下:
from flask import Flask, g, make_response, request
app = Flask(__name__)
@app.before_request
def before_request():
if "profile" in request.args:
g.profiler = Profiler()
g.profiler.start()
@app.after_request
def after_request(response):
if not hasattr(g, "profiler"):
return response
g.profiler.stop()
output_html = g.profiler.output_html()
return make_response(output_html)
這樣程序會檢測每個request中的?profile
參數,如果檢測到則會開始分析。在運行了profiler的request結束后,它會生成一個html輸出替代掉真實的response並返回。