Python 列表推导式，超级总结！

一、基本概念

列表推导式是 Python 中创建列表的一种简洁语法，它允许你在一行代码内生成列表，替代传统的 for 循环方式。其核心思想是 **"对可迭代对象中的每个元素执行某种操作，然后收集结果"**。

二、语法结构

1. 基本形式

python

[expression for item in iterable]

• expression：对每个元素执行的操作（如平方、转换类型等）
• item：迭代变量，代表可迭代对象中的每个元素
• iterable：可迭代对象（如列表、元组、range 等）

示例：生成 1 到 5 的平方列表

python

squares = [x**2 for x in range(1, 6)]
# 结果：[1, 4, 9, 16, 25]

2. 带条件过滤

python

[expression for item in iterable if condition]

• if condition：筛选条件，只处理满足条件的元素

示例：筛选偶数的平方

python

even_squares = [x**2 for x in range(1, 6) if x % 2 == 0]
# 结果：[4, 16]

3. 多重循环嵌套

python

[expression for item1 in iterable1 for item2 in iterable2]

• 等价于两层嵌套 for 循环

示例：合并两个列表的元素组合

python

pairs = [(x, y) for x in [1, 2] for y in ['a', 'b']]
# 结果：[(1, 'a'), (1, 'b'), (2, 'a'), (2, 'b')]

4. 嵌套列表推导式（二维列表）

python

[[expression for item2 in iterable2] for item1 in iterable1]

• 常用于创建矩阵或嵌套结构

示例：生成 3x3 矩阵

python

matrix = [[i*j for j in range(3)] for i in range(3)]
# 结果：[[0, 0, 0], [0, 1, 2], [0, 2, 4]]

三、与传统 for 循环的对比

传统方式：

python

squares = []
for x in range(1, 6):
    squares.append(x**2)

列表推导式：

python

squares = [x**2 for x in range(1, 6)]

优势：

• 代码更简洁、可读性更高
• 执行速度更快（底层优化）
• 减少临时变量的使用

四、高级应用

1. 条件表达式（三元运算符）

python

[expression_if_true if condition else expression_if_false for item in iterable]

示例：将正数转为平方，负数转为 0

python

numbers = [1, -2, 3, -4]
result = [x**2 if x > 0 else 0 for x in numbers]
# 结果：[1, 0, 9, 0]

2. 处理复杂数据结构

示例：从字典中提取值并转换

python

student_scores = {'Alice': 85, 'Bob': 92, 'Charlie': 78}
grades = ['Pass' if score >= 80 else 'Fail' for score in student_scores.values()]
# 结果：['Pass', 'Pass', 'Fail']

3. 生成器表达式（类似列表推导式，但返回生成器）

python

gen = (x**2 for x in range(5))  # 使用圆括号
# 节省内存，适用于大数据量处理

五、注意事项

1. 避免过度复杂：超过两层嵌套或过长的表达式会降低可读性
2. 性能权衡：对于简单操作，列表推导式更快；但涉及复杂逻辑时，传统循环可能更清晰
3. 内存占用：列表推导式会立即生成完整列表，大数据量时可能导致内存问题（考虑生成器表达式）

六、常见应用场景

1. 数据转换：如类型转换、单位换算
2. 数据筛选：过滤不符合条件的元素
3. 组合生成：生成排列组合
4. 初始化数据结构：如创建矩阵、填充默认值

通过列表推导式，我们可以用更少的代码实现复杂的数据处理逻辑，同时保持代码的简洁性和可读性。