12、推导式
如果需要生成一个长度为 100、内容为 0 的列表，需要在括号中书写 100 个 0，既繁琐又容易出错，显然是不合适的。可以使用动态的方式完成这项任务：
list = []
for i in range(100):
list.append(0)
如果需要生成一个包含 0 到 100 之间（不包括 100）所有的偶数的列表，可以在以上的代码基础上进行修改，如下所示：
list = []
for i in range(100):
if i % 2 == 0:
list.append(i)
Python 提供了列表推导的语法用于快速的构建一个列表，完成类似以上的任务
类似的，Python 提供了集合推导的语法用于快速的构建一个集合，以及字典推导的语法用于快速的构建一个字典。
列表推导
定义
注意在 for 前面的语句是循环时执行的语句
- 使用列表推导生成一个包含 4 个整数 0 的列表，代码如下：
>>> [0 for i in range(4)]
[0, 0, 0, 0]
- 使用列表推导生成一个包含 0 到 4 之间所有整数的列表，代码如下：
>>> [i for i in range(4)]
[0, 1, 2, 3]
条件语句
在循环中符合if条件语句的项目将输出for前面的表达式
使用列表推导生成一个包含 0 到 4 之间所有偶数的列表，使用 if i % 2 == 0 选择将偶数加入到列表，代码如下：
>>> [i for i in range(4) if i % 2 == 0]
[0, 2]
for 循环嵌套
第一个for循环嵌套第二个for循环，最后输出第一个for前面的表达式
使用列表推导生成一个包含两个字符的字符串列表，代码如下：
>>> [x + y for x in 'ABC' for y in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
应用举例
- 初始化长度为 10 的列表
>>> [0 for i in range(10)]
[0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
- 将列表中的元素乘以 10
>>> list = [1, 2, 3]
>>> [i*10 for i in list]
[10, 20, 30]
- 将列表中所有的字符串变成大写
>>> list = ['www', 'imooc', 'com']
>>> [i.upper() for i in list]
['WWW', 'IMOOC', 'COM']
- 包含所有的正整数的列表
>>> list = [-1, 1, -2, 2, -3, 3]
>>> [i for i in list if i > 0]
[1, 2, 3]
- 初始化生成一个 M*N 的矩阵
>>> M = 2
>>> N = 3
>>> matrix = [[0 for col in range(N)] for row in range(M)]
>>> matrix
[[0,0,0], [0,0,0]]
- 将矩阵平坦化
>>> matrix = [[1,2,3],[4,5,6]]
>>> matrix2 = [matrix[row][col] for row in range(2) for col in range(3)]
>>> matrix2
[1,2,3,4,5,6]
集合推导
应用举例
- 包含从 0 到 10 之间所有整数的集合
>>> {i for i in range(10)}
{0, 1, 2, 3, 4, 5, 6, 7, 8, 9}
- 将集合中的元素乘以 10
>>> set = {1, 2, 3}
>>> {i*10 for i in set}
{10, 20, 30}
- 将集合中所有的字符串变成大写
>>> set = {'www', 'imooc', 'com'}
>>> {i.upper() for i in set}
{'WWW', 'IMOOC', 'COM'}
- 包含所有的正整数的集合
>>> set = {-1, 1, -2, 2, -3, 3}
>>> {i for i in list if i > 0}
{1, 2, 3}
- 包含两个字符的字符串集合
>>> {x + y for x in 'ABC' for y in 'XYZ'}
{'AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ'}
字典推导
应用举例
- 创建从小写到大写的映射
>>> list = ['www', 'imooc', 'com']
>>> {i:i.upper() for i in list}
{'www': 'WWW', 'imooc': 'IMOOC', 'com': 'COM'}
- 对字典进行筛选
>>> all_students = {'tom': 95, 'jerry': 80, 'mike': 99, 'john': 70}
>>> good_students = {item[0]:item[1] for item in all_students.items() if item[1] > 90}
>>> good_students
{'tom': 95, 'mike': 99}
- 对字典进行筛选
>>> all_students = {'tom': 95, 'jerry': 80, 'mike': 99, 'john': 70}
>>> good_students = {name:grade for name,grade in all_students.items() if grade > 90}
>>> good_students
{'tom': 95, 'mike': 99}
