列表5：列表推导式实操

目录

1. 列表推导式基础

2. 复杂表达式的列表推导式

3. 带条件筛选功能的列表表达式

---

1. 列表推导式基础

列表推导式创建列表的方式更简洁。常见的用法为，对序列或可迭代对象中的每个元素应用某种操作，用生成的结果创建新的列表；或用满足特定条件的元素创建子序列。

列表推导式结构：

[expression for target in iterable]

相比for循环实现，列表推导式采取的先创建列表直接复制后替换的方法，实现不同、效果一样、效率更高。

如下两种创建平方值的列表等价：

#for 循环迭代生成
squares = []
for x in range(10):
     squares.append(x**2)

squares
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]


#列表推导式生成
squares = [x**2 for x in range(10)]
squares
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

2. 复杂表达式的列表推导式

列表推导式的方括号内包含以下内容：一个表达式，后面为一个 for 子句，然后，是零个或多个 for 或 if 子句。

结果是由表达式依据 for 和 if 子句求值计算而得出一个新列表。

如下可见两个有复杂表到时的列表推导式等价：

#for 循环迭代生成
combs = []
for x in [1,2,3]:
    for y in [3,1,4]:
        if x != y:
            combs.append((x, y))
combs
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

#列表表达式等价生成
[(x, y) for x in [1,2,3] for y in [3,1,4] if x != y]
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]

列表推导式还可以使用复杂的表达式和嵌套函数：

from math import pi
[str(round(pi, i)) for i in range(1, 6)]
['3.1', '3.14', '3.142', '3.1416', '3.14159']

3. 带条件筛选功能的列表表达式

形式如下：

[expression for target in iterable if condition1]

执行顺序为：

1. for循环语句
2. if条件判断语句
3. expression表达式

示例：

#基础示例
s = [i for i in range(10) if i % 2 == 0]
s
Out[13]: [0, 2, 4, 6, 8]

#可见先判断i是否为偶数，再加1
s = [i + 1 for i in range(10) if i % 2 == 0]
s
Out[15]: [1, 3, 5, 7, 9]