python推导式

列表推导式

• 基本格式
  variable = [out_exp_res for out_exp in input_list if out_exp == 2]
  语法说明
          out_exp_res:　　列表生成元素表达式，可以是有返回值的函数。
          for out_exp in input_list：　　迭代input_list将out_exp传入out_exp_res表达式中。
          if out_exp == 2：　　根据条件过滤哪些值可以。

>>>list = [i for i in range(10)]
>>>[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
>>>list2 = [i for i in range(10) if i%2==0]
>>>[0, 2, 4, 6, 8]
>>>import random
>>>list1 =[random.randint(0,100) for i in range(10)]
>>>[35, 70, 89, 49, 97, 49, 88, 96, 85, 12]

使用（）替代原先的[ ] 将会得到一个generator生成器对象

str = 'python'
generator = (i for i in str)
print(type(generator))
# 输出生成器对象
genIter = iter(generator)
for i in range(len(str)):
    print(next(genIter))

字典推导式

• 字典推导和列表推导的使用方法是类似的，只不中括号该改成大括号。

>>>dict = {'a': 1, 'b': 2, 'A': 3, 'Z': 4}
>>>dict1 = {
          k: dict.get(k, 0) + dict.get(k, 0)
          for k in dict.keys()
          if k in ['a','b']
            }
>>>{'a': 2, 'b': 4}
>>>dicta = {1:'apple',2:'pear',3:"banana"}
>>>dictb = { v:k for k,v in dicta.items()}
>>>{'apple': 1, 'pear': 2, 'banana': 3}

集合推导式

• 集合推导和列表推导的使用方法是类似的，只不中括号该改成大括号。

>>>set = {i for i in range(10) if i%2==0}
>>>{0, 2, 4, 6, 8}