python 推导式

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])

你可能感兴趣的:(python 推导式)