Python推导式

列表推导式

现在有个需求：30内能被3整除的数追加到一个列表

l1 = []
for i in range(1,31):
     if i % 3 ==0:
         l1.append(i)
print(l1)

可以看出来我们用了五行代码才完成这个功能，下面介绍一个简单的方法

l1 = [i for i in range(1,31) if i % 3 == 0]
print(l1)

例二：30以内所有能被3整除的数的平方

def squared(x):

    return x*x

multiples = [squared(i) for i in range(30) if i % 3 is 0]

print(multiples)

例三:找到嵌套列表中名字含有两个‘e’的所有名字

names = [['Tom', 'Billy', 'Jefferson', 'Andrew', 'Wesley', 'Steven', 'Joe'],
         ['Alice', 'Jill', 'Ana', 'Wendy', 'Jennifer', 'Sherry', 'Eva']]
print([i for l in names for i in l if i.count('e') == 2])

总结

列表推到式简单，一行搞定

特别复杂的数据列表推到式无法实现，只能用其他的方式实现

列表推到式不易排错（没办法通过debug来找出错误）

把列表解析的[]换成()得到的就是生成器表达式，例如：

l1 = (i for i in range(1,31) if i % 3 == 0)
for i in l1:
    print(i)

列表解析与生成器表达式都是一种便利的编程方式，只不过生成器表达式更节省内存

Python不但使用迭代器协议，让for循环变得更加通用。大部分内置函数，也是使用迭代器协议访问对象的。例如， sum函数是Python的内置函数，该函数使用迭代器协议访问对象，而生成器实现了迭代器协议，所以，我们可以直接这样计算一系列值的和

sum(x ** 2 for x in range(4))

列表推导式与生成器表达式区别

1 ,列推直观能看出，但是占内存

2，生成器表达式不易看出，但是节省内存。

字典推导式

例一：将一个字典的key和value对调

mcase = {'a': 10, 'b': 34}

mcase_frequency = {mcase[k]: k for k in mcase}

print(mcase_frequency)

例二：合并大小写对应的value值，将k统一成小写

mcase = {'a': 10, 'b': 34, 'A': 7, 'Z': 3}

mcase_frequency = {k.lower(): mcase.get(k.lower(), 0) + mcase.get(k.upper(), 0) for k in mcase.keys()}

print(mcase_frequency)

集合推导式

例：计算列表中每个值的平方，自带去重功能

squared = {x**2 for x in [1, -1, 2]}

print(squared)

# Output: set([1, 4])