3、python中推导式的用法

定义

推导式是Python中按照固定的条件表达式，自动生成数据的一种语法规则。

推导式是从一个或者多个迭代器快速创建序列的一种方法。它可以将循环和条件判断结合，从而避免冗长的代码。推导式是典型的 Python 风格

通过推导式，可以直接优化传统的通过循环结构和选择结构实现的列表操作，简化传统语法生成数据的规则或者过滤数据的规则。

目前，Python存在着三种推导式，分别为：列表推导式、字典推导式和集合推导式。其中，最常用的就是列表推导式。

列表推导式

语法：

[表达式foritemin 可迭代对象 ]或者：{表达式foritemin 可迭代对象if条件判断}

x = [x for x in range(1,5)]
print(x)  #[1, 2, 3, 4]

x = [x*2 for x in range(1,5)]
print(x) #[2, 4, 6, 8]

x = [x*2 for x in range(1,20) if x%5==0 ]
print(x) #[10, 20, 30]

案例1：实现数字1-10的平方

传统列表方法：

num_list = list()

for i in range(1,11):
    num_list.append(i**2)  #**表示乘方

print(num_list)
# [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

列表推导式：

num_list = [x**2 for x in range(1,11)]

print(num_list)
# [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

通过代码的对比，我们可以发现，推导式语法确实十分简洁优雅，只需要一行的代码就可以实现Python的循环结构或选择结构，基本上任何可迭代的数据都可以在推导式内进行更新。

string = 'i love python'

# 将字符串按照空格分割成一个列表
lst = [ i for i in string.split(' ')]

print(lst)
# ['i', 'love', 'python']

当我们需要一些稍显复杂的数据时，此时需要对推导式内的表达式进行嵌套，以便得到我们想要的数据。

lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] 

# 获取列表lst中的偶数
num_list = [j for i in lst for j in i if j % 2 == 0]

print(num_list)
# [2, 4, 6, 8]

由上述代码可见，一旦推导式开始嵌套，可读性便会稍微有所下降，所以，我们需要对上述代码进行优化。

lst = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

num_list = [
    j 
    for i in lst 
    for j in i 
    if j % 2 == 0
]

通过上述代码的改写，是不是直观了许多。

字典推导式

定义：

字典推导式原理上跟列表推导式其实并无不同，只需要将方括号[ ]更换为花括号{ }，换汤不换药而已。

格式：{key_expression:value_expressionfor表达式in可迭代对象}

类似于列表推导式，字典推导也可以增加 if 条件判断、多个 for 循环。

统计文本中字符出现的次数：

my_text = ' i love you, i love china, i love gao'
char_count = {c: my_text.count(c) for c in my_text}
print(char_count) 

# {' ': 9, 'i': 4, 'l': 3, 'o': 5, 'v': 3, 'e': 3, 'y': 1, 'u': 1, ',': 2, 'c': 1, 'h': 1, 'n': 1, 'a': 2, 'g': 1}

key 和 value互换： 

dct = {1: 'a', 2: 'b', 3: 'c'}

# key与value的互换
dict_list = {value: key for key, value in dct.items()}

print(dict_list)
# {'a': 1, 'b': 2, 'c': 3}

集合推导式

集合推导式和字典推导式一样，用的都是花括号{ }，在使用时，需要依靠变量和循环体对字典推导式进行区分。

语法格式：{表达式foritemin 可迭代对象 }或者：{表达式foritemin 可迭代对象if条件判断}

代码：

x = {x for x in range(1,100) if x%9==0}
print(x)
#{99, 36, 72, 9, 45, 81, 18, 54, 90, 27, 63}

推导式只是一种简化语法，不可盲目追求这种写法，毕竟传统语法的可读性往往更友好，并且，对推导式进行排错会更加困难。