相信用python的同學(xué)不少，本人也一直對python情有獨(dú)鐘，毫無疑問python作為一門解釋性動態(tài)語言沒有那些編譯型語言高效，但是python簡潔、易讀以及可擴(kuò)展性等特性使得它大受青睞。

 工作中很多同事都在用python，但往往很少有人關(guān)注它的性能和慣用法，一般都是現(xiàn)學(xué)現(xiàn)用，畢竟python不是我們的主要語言，我們一般只是使用它來做一些系統(tǒng)管理的工作。但是我們?yōu)槭裁床蛔龅母媚?？python zen中有這樣一句：There should be one-- and preferably only one --obvious way to do it. Although that way may not be obvious at first unless you're Dutch. 大意就是python鼓勵使用一種最優(yōu)的方法去完成一件事，這也是和ruby等的一個(gè)差異。所以一種好的python編寫習(xí)慣個(gè)人認(rèn)為很重要，本文就重點(diǎn)從性能角度出發(fā)對python的一些慣用法做一個(gè)簡單總結(jié)，希望對大家有用~

    提到性能，最容易想到的是降低復(fù)雜度，一般可以通過測量代碼回路復(fù)雜度（cyclomatic complexitly）和Landau符號（大O）來分析, 比如dict查找是O(1)，而列表的查找卻是O(n)，顯然數(shù)據(jù)的存儲方式選擇會直接影響算法的復(fù)雜度。

一、數(shù)據(jù)結(jié)構(gòu)的選擇
1. 在列表中查找：

 對于已經(jīng)排序的列表考慮用bisect模塊來實(shí)現(xiàn)查找元素，該模塊將使用二分查找實(shí)現(xiàn)

def find(seq, el) :
  pos = bisect(seq, el)
  if pos == 0 or ( pos == len(seq) and seq[-1] != el ) :
    return -1
  return pos - 1

而快速插入一個(gè)元素可以用:

 bisect.insort(list, element) 

這樣就插入元素并且不需要再次調(diào)用 sort() 來保序，要知道對于長list代價(jià)很高.

2. set代替列表：

 比如要對一個(gè)list進(jìn)行去重，最容易想到的實(shí)現(xiàn)：

seq = ['a', 'a', 'b']
res = []
for i in seq:
  if i not in res:
    res.append(i)

顯然上面的實(shí)現(xiàn)的復(fù)雜度是O(n2)，若改成：

seq = ['a', 'a', 'b']
res = set(seq)

復(fù)雜度馬上降為O(n)，當(dāng)然這里假定set可以滿足后續(xù)使用。

另外，set的union，intersection，difference等操作要比列表的迭代快的多，因此如果涉及到求列表交集，并集或者差集等問題可以轉(zhuǎn)換為set來進(jìn)行，平時(shí)使用的時(shí)候多注意下，特別當(dāng)列表比較大的時(shí)候，性能的影響就更大。

3. 使用python的collections模塊替代內(nèi)建容器類型：

collections有三種類型：

deque：增強(qiáng)功能的類似list類型
defaultdict：類似dict類型
namedtuple：類似tuple類型

       列表是基于數(shù)組實(shí)現(xiàn)的，而deque是基于雙鏈表的，所以后者在中間or前面插入元素，或者刪除元素都會快很多。

       defaultdict為新的鍵值添加了一個(gè)默認(rèn)的工廠，可以避免編寫一個(gè)額外的測試來初始化映射條目，比dict.setdefault更高效，引用python文檔的一個(gè)例子：

#使用profile stats工具進(jìn)行性能分析

>>> from pbp.scripts.profiler import profile, stats
>>> s = [('yellow', 1), ('blue', 2), ('yellow', 3),
... ('blue', 4), ('red', 1)]
>>> @profile('defaultdict')
... def faster():
... d = defaultdict(list)
... for k, v in s:
... d[k].append(v)
...
>>> @profile('dict')
... def slower():
... d = {}
... for k, v in s:
... d.setdefault(k, []).append(v)
...
>>> slower(); faster()
Optimization: Solutions
[ 306 ]
>>> stats['dict']
{'stones': 16.587882671716077, 'memory': 396,
'time': 0.35166311264038086}
>>> stats['defaultdict']
{'stones': 6.5733464259021686, 'memory': 552,
'time': 0.13935494422912598}

可見性能提升了快3倍。defaultdict用一個(gè)list工廠作為參數(shù)，同樣可用于內(nèi)建類型，比如long等。

除了實(shí)現(xiàn)的算法、架構(gòu)之外，python提倡簡單、優(yōu)雅。所以正確的語法實(shí)踐又很有必要，這樣才會寫出優(yōu)雅易于閱讀的代碼。

二、語法最佳實(shí)踐
字符串操作：優(yōu)于python字符串對象是不可改變的，因此對任何字符串的操作如拼接，修改等都將產(chǎn)生一個(gè)新的字符串對象，而不是基于原字符串，因此這種持續(xù)的 copy會在一定程度上影響Python的性能：
        (1)用join代替 '+' 操作符，后者有copy開銷；

        (2)同時(shí)當(dāng)對字符串可以使用正則表達(dá)式或者內(nèi)置函數(shù)來處理的時(shí)候，選擇內(nèi)置函數(shù)。如str.isalpha()，str.isdigit()，str.startswith((‘x', ‘yz'))，str.endswith((‘x', ‘yz'))

        (3)字符格式化操作優(yōu)于直接串聯(lián)讀取：

     str = "%s%s%s%s" % (a, b, c, d)  # efficient
     str = "" + a + b + c + d + ""  # slow

2. 善用list comprehension(列表解析)  & generator（生成器） & decorators（裝飾器），熟悉itertools等模塊：

(1) 列表解析，我覺得是python2中最讓我印象深刻的特性，舉例1：

   >>> # the following is not so Pythonic 
   >>> numbers = range(10)
   >>> i = 0 
   >>> evens = [] 
   >>> while i < len(numbers): 
   >>>  if i %2 == 0: evens.append(i) 
   >>>  i += 1 
   >>> [0, 2, 4, 6, 8] 

   >>> # the good way to iterate a range, elegant and efficient
   >>> evens = [ i for i in range(10) if i%2 == 0] 
   >>> [0, 2, 4, 6, 8]  

舉例2：

def _treament(pos, element):
  return '%d: %s' % (pos, element)
f = open('test.txt', 'r')
if __name__ == '__main__':
  #list comps 1
  print sum(len(word) for line in f for word in line.split())
  #list comps 2
  print [(x + 1, y + 1) for x in range(3) for y in range(4)]
  #func
  print filter(lambda x: x % 2 == 0, range(10))
  #list comps3
  print [i for i in range(10) if i % 2 == 0]
  #list comps4 pythonic
  print [_treament(i, el) for i, el in enumerate(range(10))]

output：
24
[(1, 1), (1, 2), (1, 3), (1, 4), (2, 1), (2, 2), (2, 3), (2, 4), (3, 1), (3, 2), (3, 3), (3, 4)]
[0, 2, 4, 6, 8]
[0, 2, 4, 6, 8]
['0: 0', '1: 1', '2: 2', '3: 3', '4: 4', '5: 5', '6: 6', '7: 7', '8: 8', '9: 9']

沒錯(cuò)，就是這么優(yōu)雅簡單。

   (2) 生成器表達(dá)式在python2.2引入，它使用'lazy evaluation'思想，因此在使用內(nèi)存上更有效。引用python核心編程中計(jì)算文件中最長的行的例子：

f = open('/etc/motd, 'r')
longest = max(len(x.strip()) for x in f)
f.close()
return longest

這種實(shí)現(xiàn)簡潔而且不需要把文件文件所有行讀入內(nèi)存。

 (3) python在2.4引入裝飾器，又是一個(gè)讓人興奮的特性，簡單來說它使得函數(shù)和方法封裝（接收一個(gè)函數(shù)并返回增強(qiáng)版本的函數(shù)）更容易閱讀、理解。'@'符號是裝飾器語法，你可以裝飾一個(gè)函數(shù)，記住調(diào)用結(jié)果供后續(xù)使用，這種技術(shù)被稱為memoization的，下面是用裝飾器完成一個(gè)cache功能：

import time
import hashlib
import pickle
from itertools import chain
cache = {}
def is_obsolete(entry, duration):
  return time.time() - entry['time'] > duration

def compute_key(function, args, kw):
  #序列化/反序列化一個(gè)對象,這里是用pickle模塊對函數(shù)和參數(shù)對象進(jìn)行序列化為一個(gè)hash值
  key = pickle.dumps((function.func_name, args, kw))
  #hashlib是一個(gè)提供MD5和sh1的一個(gè)庫，該結(jié)果保存在一個(gè)全局字典中
  return hashlib.sha1(key).hexdigest()

def memoize(duration=10):
  def _memoize(function):
    def __memoize(*args, **kw):
      key = compute_key(function, args, kw)

      # do we have it already
      if (key in cache and
        not is_obsolete(cache[key], duration)):
        print 'we got a winner'
        return cache[key]['value']

      # computing
      result = function(*args, **kw)
      # storing the result
      cache[key] = {'value': result,-
              'time': time.time()}
      return result
    return __memoize
  return _memoize

@memoize()
def very_very_complex_stuff(a, b, c):
  return a + b + c

print very_very_complex_stuff(2, 2, 2)
print very_very_complex_stuff(2, 2, 2)


@memoize(1)
def very_very_complex_stuff(a, b):
  return a + b

print very_very_complex_stuff(2, 2)
time.sleep(2)
print very_very_complex_stuff(2, 2)

運(yùn)行結(jié)果：

6

we got a winner

6

4

4

裝飾器在很多場景用到，比如參數(shù)檢查、鎖同步、單元測試框架等，有興趣的人可以自己進(jìn)一步學(xué)習(xí)。

3.  善用python強(qiáng)大的自省能力（屬性和描述符）：自從使用了python，真的是驚訝原來自省可以做的這么強(qiáng)大簡單，關(guān)于這個(gè)話題，限于內(nèi)容比較多，這里就不贅述，后續(xù)有時(shí)間單獨(dú)做一個(gè)總結(jié)，學(xué)習(xí)python必須對其自省好好理解。

三、 編碼小技巧
1、在python3之前版本使用xrange代替range，因?yàn)閞ange()直接返回完整的元素列表而xrange()在序列中每次調(diào)用只產(chǎn)生一個(gè)整數(shù)元素，開銷小。（在python3中xrange不再存在，里面range提供一個(gè)可以 遍歷任意長度的范圍的iterator）
2、if done is not None比語句if done != None更快;
3、盡量使用"in"操作符，簡潔而快速: for i in seq: print i
4、'x < y < z'代替'x < y and y < z'；
5、while 1要比while True更快, 因?yàn)榍罢呤菃尾竭\(yùn)算，后者還需要計(jì)算；
6、盡量使用build-in的函數(shù)，因?yàn)檫@些函數(shù)往往很高效，比如add(a,b)要優(yōu)于a+b;
7、在耗時(shí)較多的循環(huán)中，可以把函數(shù)的調(diào)用改為內(nèi)聯(lián)的方式，內(nèi)循環(huán)應(yīng)該保持簡潔。
8、使用多重賦值來swap元素：

      x, y = y, x  # elegant and efficient

 而不是：

      temp = x
      x = y
      y = temp 

9. 三元操作符（python2.5后）：V1 if X else V2，避免使用(X and V1) or V2，因?yàn)楹笳弋?dāng)V1=""時(shí)，就會有問題。

10. python之switch case實(shí)現(xiàn)：因?yàn)閟witch case語法完全可用if else代替，所以python就沒  有switch case語法，但是我們可以用dictionary或lamda實(shí)現(xiàn)：

switch case結(jié)構(gòu)：

switch (var)
{
  case v1: func1();
  case v2: func2();
  ...
  case vN: funcN();
  default: default_func();
}
dictionary實(shí)現(xiàn)：

values = {
      v1: func1,
      v2: func2,
      ...
      vN: funcN,
     }
values.get(var, default_func)()
lambda實(shí)現(xiàn)：

{
 '1': lambda: func1,
 '2': lambda: func2,
 '3': lambda: func3
}[value]()

用try…catch來實(shí)現(xiàn)帶Default的情況，個(gè)人推薦使用dict的實(shí)現(xiàn)方法。

 這里只總結(jié)了一部分python的實(shí)踐方法，希望這些建議可以幫助到每一位使用python的同學(xué)，優(yōu)化性能不是重點(diǎn)，高效解決問題，讓自己寫的代碼更加易于維護(hù)！

最新評論