python之推導式介紹

  今天在看公司的批量部署代碼的時候（使用的fabric），發現一個頗有意思的裝飾器，裏面用到了python的推導式，特別記錄一下。

代碼以下：

def host_restrict(tags=[], names=[]):
    def decorator(f):
        @functools.wraps(f)
        def decorated(*args, **kwargs):
            host_info = hosts.get(env.host)
            if not host_info:
                return f(*args, **kwargs)
            if tags and host_info[0] not in tags:
                raise StandardError('Tag %s not in %s' % (host_info[0], tags))
            if names and not set(names).intersection(
                    set([s for s in host_info[1].split(' ') if s.strip()])):
                raise StandardError(
                    'Name %s not in %s' % (host_info[1], names))
            return f(*args, **kwargs)
        return decorated
    return decorator

備註: 這個裝飾器主要是用來判斷用戶輸入的tag和names 是否符合要求，例如:

#host_info的格式爲
host_info = ['api','nginx']

@host_restrict(['api'], ['nginx'])
def init():
    ...

   好了，不關注細節，主要有一行代碼引發了個人注意 :

 set([s for s in host_info[1].split(' ') if s.strip()])

這就是一個 python的列表推導式。

推導式是python中很強大的、很受歡迎的特性，具備語言簡潔，速度快等優勢。推導式包括:

1. 列表推導式

2. 字典推導式

3. 集合推導式

4. 嵌套列表推導式

NOTE: 字典和集合推導是最近才加入到Python的(Python 2.7 和Python 3.1以上版). 

下面簡要介紹下:

    【列表推導式】

  列表推導能很是簡潔的構造一個新列表:只用一條簡潔的表達式便可對獲得的元素進行轉換變形

  其基本格式以下:

[expr for value in collection if condition]

過濾條件無關緊要，取決於實際應用，只留下表達式，至關於下面這段for循環:

result = []
forvalue in collection:
    ifcondition:
       result.append(expression)

 示例代碼：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#例1: 過濾掉長度小於3的字符串列表，並將剩下的轉換成大寫字母
names = ['Bob','Tom','alice','Jerry','Wendy','Smith']
print [name.upper() for name in names if len(name) > 3]
#例2: 打印出那些能被3整除的平方數
print [x*x for x in range(10) if x%3==0]

 輸出：

['ALICE', 'JERRY', 'WENDY', 'SMITH']
[0, 9, 36, 81]

    

    【字典推導式】

  字典和集合推導式是該思想的延續，語法差很少，只不過產生的是集合和字典而已。其基本格式以下:

{ key_expr:value_expr for value in collection if condition }

 示例代碼：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#例1: 用字典推導式 -- 以字符串以及其長度建字典
strings = ['import','is','with','if','file','exception']
print {key:len(key) for val,key in enumerate(strings)}

 輸出：

 {'exception': 9, 'is': 2, 'file': 4, 'import': 6, 'with': 4, 'if': 2}

    

    【集合推導式】

  集合推導式跟列表推導式很是類似，惟一區別在於用{}代替[]。其基本格式以下:

{ expr for value incollection if condition }

 示例代碼：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
#例1: 用集合推導建字符串長度的集合
strings = ['a','is','with','if','file','exception']
print {len(s) for s in strings}

 輸出：

set([1, 2, 4, 9])

    

    【嵌套列表推導式】

  嵌套列表是指列表中嵌套列表，好比說:

L = [[1,2,3],
         [4,5,6],
         [7,8,9]]

  示例代碼：

#例1: list comprehension替代嵌套循環
print [[x,y] for x in range(3) for y in range(6)]
#至關於
z = []
for x in range(3):
    for y in range(6):
        z.append([x,y])
print z
#例2: 一個由男人列表和女人列表組成的嵌套列表,取出姓名中帶有兩個以上字母e的姓名，組成列表

#用for循環實現:
tmp = []
names = [
    ['Tom','Billy','Jefferson','Andrew','Wesley','Steven','Joe'],
    ['Alice','Jill','Ana','Wendy','Jennifer','Sherry','Eva']
]
for lst in names:
    for name in lst:
        if name.count('e') >= 2:
            tmp.append(name)
print tmp
#用嵌套列表實現:
print [name for lst in names for name in lst if name.count('e') >= 2]

輸出：

['Jefferson', 'Wesley', 'Steven', 'Jennifer']
['Jefferson', 'Wesley', 'Steven', 'Jennifer']

 

還有一些比較好的文章：

http://blog.chinaunix.net/uid-28631822-id-3488324.html

http://python.jobbole.com/80823/