推导式comprehensions(又称解析式),是Python的一种独有特性。推导式是可以从一个数据序列构建另一个新的数据序列的结构体。共有三种推导,在Python2和Python3中都有支持:
- 列表(list)推导式
- 字典(dict)推导式
- 集合(set)推导式
列表推导式
使用[]生成list
例一:
result = [i for i in range(30) if i % 3 is 0]
print('result = ',result)
# result = [0, 3, 6, 9, 12, 15, 18, 21, 24, 27]
例二:
def squared(x):
return x*x
result = [squared(i) for i in range(30) if i % 3 is 0]
print('result = ',result)
# result = [0, 9, 36, 81, 144, 225, 324, 441, 576, 729]
使用()生成generator
将两表推导式的[]改成()即可得到生成器
result = (i for i in range(30) if i % 3 is 0)
print('result = ',result)
# result = <generator object <genexpr> at 0x0000000002B915C8>
字典推导式
字典推导和列表推导的使用方法是类似的,只不过中括号改成大括号。
**例一:**大小写key合并
mcase = {'a':10,'b':34,'A':7,'Z':3}
result = {
k.lower():mcase.get(k.lower(),0) + mcase.get(k.upper(),0)
for k in mcase.keys()
if k.lower() in ['a','b']
}
print(result) # {'a': 17, 'b': 34}
**例二:**快速更换key和value
mcase = {'a':10,'b':34}
result = {v:k for k ,v in mcase.items()}
print(result) # {10: 'a', 34: 'b'}
集合推导式
它们跟列表推导式也是类似的,唯一的区别在于它使用大括号
result = {x**2 for x in [1,1,2]}
print(result) # {1, 4}