Python 推导式

​

 推导式是Python中很强大的、很受欢迎的特性，具有语言简洁，速度快等优点。推导式包括:

1.列表推导式
2.字典推导式
3.集合推导式

【列表推导式】

列表推导能非常简洁的构造一个新列表:只用一条简洁的表达式即可对得到的元素进行转换变形
其基本格式如下:

[expr for value in collection ifcondition]

[表达式 for 变量 in 列表]    或者  [表达式 for 变量 in 列表 if 条件]

 

过滤条件可有可无，取决于实际应用，只留下表达式；相当于下面这段for循环:
 

result = []  

for value in collection:  

    if condition:  

        result.append(expression) 

 

例: 

#!/usr/bin/python
# -*- coding: utf-8 -*-
li = [1,2,3,4,5,6,7,8,9]
print [x**2 for x in li]
print [x**2 for x in li if x>5]
print dict([(x,x*10) for x in li])

print  [ (x, y) for x in range(10) if x % 2 if x > 3 for y in range(10) if y > 7 if y != 8 ]
vec=[2,4,6]
vec2=[4,3,-9]
sq = [vec[i]+vec2[i] for i in range(len(vec))]
print sq
print [x*y for x in [1,2,3] for y in  [1,2,3]]
testList = [1,2,3,4]
def mul2(x):
    return x*2
print [mul2(i) for i in testList]

 

 

【字典推导式】

字典和集合推导式是该思想的延续，语法差不多，只不过产生的是集合和字典而已。其基本格式如下:

{ key_expr: value_expr for value in collection if condition }

例1:  用字典推导式以字符串以及其长度建字典 

>>> strings = ['import','is','with','if','file','exception']  

  

>>> D = {key: val for val,key in enumerate(strings)}  

  

>>> D  

{'exception': 5, 'is': 1, 'file': 4, 'import': 0, 'with': 2, 'if': 3}  

【集合推导式】
集合推导式跟列表推导式非常相似，唯一区别在于用{}代替[]。其基本格式如下:

{ expr for value in collection if condition }

 

例1:  用集合推导建字符串长度的集合
 

>>> strings = ['a','is','with','if','file','exception']  

>>> {len(s) for s in strings}    #有长度相同的会只留一个，这在实际上也非常有用  

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