01 Python 為什麼效能差
當我們提到一門程式語言的效率時:通常有兩層意思,第一是開發效率,這是對程式設計師而言,完成編碼所需要的時間;另一個是執行效率,這是對計算機而言,完成計算任務所需要的時間。編碼效率和執行效率往往是魚與熊掌的關係,是很難同時兼顧的。不同的語言會有不同的側重,python語言毫無疑問更在乎編碼效率,life is short,we use python。
雖然使用python的程式設計人員都應該接受其執行效率低的事實,但python在越多越來的領域都有廣泛應用,比如科學計算 、web伺服器等。程式設計師當然也希望python能夠運算得更快,希望python可以更強大。
首先,python相比其他語言具體有多慢,這個不同場景和測試用例,結果肯定是不一樣的。這個網址給出了不同語言在各種case下的效能對比,這一頁是python3和C++的對比,下面是兩個case:
從上圖可以看出,不同的case,python比C++慢了幾倍到幾十倍。
Python運算效率低,具體是什麼原因呢,下列羅列一些
Python是動態語言
一個變數所指向物件的型別在執行時才確定,編譯器做不了任何預測,也就無從優化。舉一個簡單的例子: r = a + b。 a和b相加,但a和b的型別在執行時才知道,對於加法操作,不同的型別有不同的處理,所以每次執行的時候都會去判斷a和b的型別,然後執行對應的操作。而在靜態語言如C++中,編譯的時候就確定了執行時的程式碼。
另外一個例子是屬性查詢,關於具體的查詢順序在《python屬性查詢》中有詳細介紹。簡而言之,訪問物件的某個屬性是一個非常複雜的過程,而且通過同一個變數訪問到的python物件還都可能不一樣(參見Lazy property的例子)。而在C語言中,訪問屬性用物件的地址加上屬性的偏移就可以了。
Python是解釋執行
但是不支援JIT(just in time compiler)。雖然大名鼎鼎的google曾經嘗試Unladen Swallow 這個專案,但最終也折了。
Python中一切都是物件
每個物件都需要維護引用計數,增加了額外的工作。
Python GIL
GIL是Python最為詬病的一點,因為GIL,python中的多執行緒並不能真正的併發。如果是在IO bound的業務場景,這個問題並不大,但是在CPU BOUND的場景,這就很致命了。所以筆者在工作中使用python多執行緒的情況並不多,一般都是使用多程式(pre fork),或者在加上協程。即使在單執行緒,GIL也會帶來很大的效能影響,因為python每執行100個opcode(預設,可以通過sys.setcheckinterval()設定)就會嘗試執行緒的切換,具體的原始碼在ceval.c::PyEval_EvalFrameEx。
垃圾回收
這個可能是所有具有垃圾回收的程式語言的通病。python採用標記和分代的垃圾回收策略,每次垃圾回收的時候都會中斷正在執行的程式(stop the world),造成所謂的頓卡。infoq上有一篇文章,提到禁用Python的GC機制後,Instagram效能提升了10%。感興趣的讀者可以去細讀。
02 程式碼Pythonic
我們都知道 過早的優化是罪惡之源,一切優化都需要基於profile。
但是,作為一個python開發者應該要Pythonic,而且pythonic的程式碼往往比non-pythonic的程式碼效率高一些,比如:
使用迭代器iterator,for example:
dict的iteritems 而不是items(同itervalues,iterkeys)
使用generator,特別是在迴圈中可能提前break的情況判斷是否是同一個物件使用 is 而不是 ==
判斷一個物件是否在一個集合中,使用set而不是list
利用短路求值特性,把“短路”概率過的邏輯表示式寫在前面。其他的lazy ideas也是可以的
對於大量字串的累加,使用join操作
使用for else(while else)語法
交換兩個變數的值使用: a, b = b, a
03 基於profile優化
即使我們的程式碼已經非常pythonic了,但可能執行效率還是不能滿足預期。我們也知道80/20定律,絕大多數的時間都耗費在少量的程式碼片段裡面了,優化的關鍵在於找出這些瓶頸程式碼。方式很多:到處加log列印時間戳、或者將懷疑的函式使用timeit進行單獨測試,但最有效的是使用profile工具。
▍ 1. python profilers
對於python程式,比較出名的profile工具有三個:profile、cprofile和hotshot。
其中profile是純python語言實現的,Cprofile將profile的部分實現native化,hotshot也是C語言實現,hotshot與Cprofile的區別在於:hotshot對目的碼的執行影響較小,代價是更多的後處理時間,而且hotshot已經停止維護了。
需要注意的是,profile(Cprofile hotshot)只適合單執行緒的python程式。對於多執行緒,可以使用yappi,yappi不僅支援多執行緒,還可以精確到CPU時間
對於協程(greenlet),可以使用greenletprofiler,基於yappi修改,用greenlet context hook住thread context
下面給出一段編造的”效率低下“的程式碼,並使用Cprofile來說明profile的具體方法以及我們可能遇到的效能瓶頸。
# -*- coding: UTF-8 -*-
from cProfile import Profile
import math
def foo():
return foo1()
def foo1():
return foo2()
def foo2():
return foo3()
def foo3():
return foo4()
def foo4():
return "this call tree seems ugly, but it always happen"
def bar():
ret = 0
for i in xrange(10000):
ret += i * i + math.sqrt(i)
return ret
def main():
for i in range(100000):
if i % 10000 == 0:
bar()
else:
foo()
if __name__ == '__main__':
prof = Profile()
prof.runcall(main)
prof.print_stats()
#prof.dump_stats('test.prof') # dump profile result to test.prof
複製程式碼
執行結果如下:
對於上面的的輸出,每一個欄位意義如下:
ncalls 函式總的呼叫次數
tottime 函式內部(不包括子函式)的佔用時間
percall (第一個) tottime/ncalls
cumtime 函式包括子函式所佔用的時間
percall (第二個)cumtime/ncalls
filename:lineno(function) 檔案:行號(函式)
複製程式碼
程式碼中的輸出非常簡單,事實上可以利用pstat,讓profile結果的輸出多樣化,具體可以參見官方文件:https://docs.python.org/2/library/profile.html
▍2. profile GUI tools
雖然Cprofile的輸出已經比較直觀,但我們還是傾向於儲存profile的結果,然後用圖形化的工具來從不同的維度來分析,或者比較優化前後的程式碼。
檢視profile結果的工具也比較多,比如,visualpytune、qcachegrind、runsnakerun,本文用visualpytune做分析。對於上面的程式碼,按照註釋生成修改後重新執行生成test.prof檔案,用visualpytune直接開啟就可以了,如下:
欄位的意義與文字輸出基本一致,不過便捷性可以點選欄位名排序。左下方列出了當前函式的calller(呼叫者),右下方是當前函式內部與子函式的時間佔用情況。上如是按照cumtime(即該函式內部及其子函式所佔的時間和)排序的結果。
造成效能瓶頸的原因通常是高頻呼叫的函式、單次消耗非常高的函式、或者二者的結合。在我們前面的例子中,foo就屬於高頻呼叫的情況,bar屬於單次消耗非常高的情況,這都是我們需要優化的重點。
python-profiling-tools中介紹了qcachegrind和runsnakerun的使用方法,這兩個colorful的工具比visualpytune強大得多。具體的使用方法請參考原文,下圖給出test.prof用qcachegrind開啟的結果
qcachegrind確實要比visualpytune強大。從上圖可以看到,大致分為三部:。第一部分同visualpytune類似,是每個函式佔用的時間,其中Incl等同於cumtime, Self等同於tottime。第二部分和第三部分都有很多標籤,不同的標籤標示從不同的角度來看結果,如圖上所以,第三部分的“call graph”展示了該函式的call tree幷包含每個子函式的時間百分比,一目瞭然。
▍3. profile 針對優化
知道了熱點,就可以進行鍼對性的優化,而這個優化往往根具體的業務密切相關,沒用萬能鑰匙,具體問題,具體分析。個人經驗而言,最有效的優化是找產品經理討論需求,可能換一種方式也能滿足需求,少者稍微折衷一下產品經理也能接受。次之是修改程式碼的實現,比如之前使用了一個比較通俗易懂但效率較低的演算法,如果這個演算法成為了效能瓶頸,那就考慮換一種效率更高但是可能難理解的演算法、或者使用dirty Flag模式。對於這些同樣的方法,需要結合具體的案例,本文不做贅述。
接下來結合python語言特性,介紹一些讓python程式碼不那麼pythonic,但可以提升效能的一些做法
第一:減少函式的呼叫層次
每一層函式呼叫都會帶來不小的開銷,特別對於呼叫頻率高,但單次消耗較小的calltree,多層的函式呼叫開銷就很大,這個時候可以考慮將其展開。
對於之前調到的profile的程式碼,foo這個call tree非常簡單,但頻率高。修改程式碼,增加一個plain_foo()函式, 直接返回最終結果,關鍵輸出如下:
跟之前的結果對比:
可以看到,優化了差不多3倍。
第二:優化屬性查詢
上面提到,python 的屬性查詢效率很低,如果在一段程式碼中頻繁訪問一個屬性(比如for迴圈),那麼可以考慮用區域性變數代替物件的屬性。
第三:關閉GC
在本文的第一章節已經提到,關閉GC可以提升python的效能,GC帶來的頓卡在實時性要求比較高的應用場景也是難以接受的。但關閉GC並不是一件容易的事情。我們知道python的引用計數只能應付沒有迴圈引用的情況,有了迴圈引用就需要靠GC來處理。在python語言中, 寫出迴圈引用非常容易。比如:
# case 1
a, b = SomeClass(), SomeClass()
a.b, b.a = b, a
# case 2
lst = []
lst.append(lst)
# case 3
self.handler = self.some_func
複製程式碼
當然,大家可能說,誰會這麼傻,寫出這樣的程式碼,是的,上面的程式碼太明顯,當中間多幾個層級之後,就會出現“間接”的迴圈應用。在python的標準庫 collections裡面的OrderedDict就是case2:
要解決迴圈引用,第一個辦法是使用弱引用(weakref),第二個是手動解迴圈引用。
第四:setcheckinterval
如果程式確定是單執行緒,那麼修改checkinterval為一個更大的值,這裡有介紹。
第五:使用__slots__
slots最主要的目的是用來節省記憶體,但是也能一定程度上提高效能。我們知道定義了__slots__的類,對某一個例項都會預留足夠的空間,也就不會再自動建立__dict__。當然,使用__slots__也有許多注意事項,最重要的一點,繼承鏈上的所有類都必須定義__slots__,python doc有詳細的描述。下面看一個簡單的測試例子:
class BaseSlots(object):
__slots__ = ['e', 'f', 'g']
class Slots(BaseSlots):
__slots__ = ['a', 'b', 'c', 'd']
def __init__(self):
self.a = self.b = self.c = self.d = self.e = self.f = self.g = 0
class BaseNoSlots(object):
pass
class NoSlots(BaseNoSlots):
def __init__(self):
super(NoSlots,self).__init__()
self.a = self.b = self.c = self.d = self.e = self.f = self.g = 0
def log_time(s):
begin = time.time()
for i in xrange(10000000):
s.a,s.b,s.c,s.d, s.e, s.f, s.g
return time.time() - begin
if __name__ == '__main__':
print 'Slots cost', log_time(Slots())
print 'NoSlots cost', log_time(NoSlots())
複製程式碼
輸出結果如下
Slots cost 3.12999987602
NoSlots cost 3.48100018501
複製程式碼
-END-
關注公眾號瞭解更多技術
免費領取資料可加群,705673780