python中可以使用推导式来通过已有的数据序列快速的创建一个数据序列。目前pyhton中有三种推到式。
- 列表
- 字典
- 集合
列表推导式
variable = [元素结果 for i in 已存在的list if 判断条件]
元素结果可以为函数,也可以为表达式,也可以为值。
例如:
输出值:
temp = [i for i in range(30) if i%3==0]
print(temp)//[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]
表达式:
temp = [i**2 for i in range(30) if i%3==0]
print(temp)//[0, 9, 36, 81, 144, 225, 324, 441, 576, 729]
函数:
def iii(i):
i=i**3
return i
temp = [iii(i) for i in range(30) if i%3==0]
print(temp)//[0, 27, 216, 729, 1728, 3375, 5832, 9261, 13824, 19683]
字典推导式
variable = [k:结果值 for k in 已存在的dict if 判断条件]
variable = [k,v for k,v in 已存在的dict.items() if 判断条件]
输出值:
dict1 = {'a':1,'b':2,'c':3}
temp ={ k:dict1.get(k,0) for k in dict1 if dict1.get(k,0)%3 ==0}
print temp//{'c': 3}
表达式:
dict1 = {'a':1,'b':2,'c':3}
temp ={ k*2:dict1.get(k,0)**2 for k in dict1 if dict1.get(k,0)%3 ==0}
print temp//{'cc': 9}
函数:
def iii(i):
i=i**3;
return i
def ii(i):
i=i*2
return i
dict1 = {'a':1,'b':2,'c':3}
temp ={ ii(k):iii(dict1.get(k,0)) for k in dict1 if dict1.get(k,0)%3 ==0}
print temp//{'cc': 27}
k,v值互换
dict1 = {'a':1,'b':2,'c':3}
temp ={ v:k for k,v in dict1.items() }
print temp//{1: 'a', 2: 'b', 3: 'c'}
集合
集合推导式与列表相似,只需把[]换成{}
例子
set1 = {1,1,1,2}
temp = {i for i in set1 if i-1>=0}
print(temp)//set([1, 2])