Python编程实战 · 基础入门篇 | 推导式(列表推导式 / 字典推导式)

在学习了循环与条件判断后,你可能已经编写过很多这样的代码:

python 复制代码
result = []
for i in range(1, 6):
    result.append(i * i)
print(result)

输出:

csharp 复制代码
[1, 4, 9, 16, 25]

没错,这段代码实现了平方数列表,但显得冗长。 在 Python 中,有一种更优雅 、更高效 的写法------推导式(Comprehension)。


一、什么是推导式

推导式是一种简洁的序列生成语法,用一行代码即可实现循环、条件判断和数据构建。

Python 支持多种推导式:

类型 作用 示例
列表推导式 快速生成列表 [x*x for x in range(5)]
字典推导式 快速生成字典 {x: x*x for x in range(5)}
集合推导式 快速生成集合 {x*x for x in range(5)}
生成器表达式 生成迭代器 (x*x for x in range(5))

本章重点讲解 列表推导式 与 字典推导式。


二、列表推导式(List Comprehension)

1. 基本语法

python 复制代码
[表达式 for 变量 in 可迭代对象]

相当于:

python 复制代码
result = []
for 变量 in 可迭代对象:
    result.append(表达式)

2. 示例:生成平方数列表

python 复制代码
squares = [x * x for x in range(1, 6)]
print(squares)

输出:

csharp 复制代码
[1, 4, 9, 16, 25]

相比传统循环,代码更加简洁清晰。


3. 添加条件判断

可以在推导式中加入条件过滤,只保留满足条件的元素。

python 复制代码
even_squares = [x * x for x in range(1, 10) if x % 2 == 0]
print(even_squares)

输出:

csharp 复制代码
[4, 16, 36, 64]

✅ 这相当于"先循环,再筛选",写法紧凑但逻辑清晰。


4. 嵌套循环

推导式支持嵌套循环(类似双层 for 循环)。

python 复制代码
pairs = [(x, y) for x in [1, 2, 3] for y in ['A', 'B']]
print(pairs)

输出:

css 复制代码
[(1, 'A'), (1, 'B'), (2, 'A'), (2, 'B'), (3, 'A'), (3, 'B')]

相当于:

python 复制代码
pairs = []
for x in [1, 2, 3]:
    for y in ['A', 'B']:
        pairs.append((x, y))

5. 条件 + 嵌套

推导式可以在嵌套循环中加入条件:

python 复制代码
pairs = [(x, y) for x in range(1, 4) for y in range(1, 4) if x != y]
print(pairs)

输出:

css 复制代码
[(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)]

6. 字符串处理案例

将一句话中的单词全部转换为大写:

python 复制代码
text = "python is powerful"
upper_words = [word.upper() for word in text.split()]
print(upper_words)

输出:

css 复制代码
['PYTHON', 'IS', 'POWERFUL']

三、字典推导式(Dict Comprehension)

字典推导式与列表推导式类似,只不过生成的是键值对结构。

1. 基本语法

python 复制代码
{键表达式: 值表达式 for 变量 in 可迭代对象}

2. 示例:数字与平方的映射

python 复制代码
squares = {x: x * x for x in range(1, 6)}
print(squares)

输出:

yaml 复制代码
{1: 1, 2: 4, 3: 9, 4: 16, 5: 25}

3. 添加条件过滤

python 复制代码
even_squares = {x: x * x for x in range(1, 10) if x % 2 == 0}
print(even_squares)

输出:

yaml 复制代码
{2: 4, 4: 16, 6: 36, 8: 64}

4. 翻转字典的键与值

python 复制代码
data = {"A": 1, "B": 2, "C": 3}
reversed_dict = {v: k for k, v in data.items()}
print(reversed_dict)

输出:

css 复制代码
{1: 'A', 2: 'B', 3: 'C'}

5. 实战:学生成绩筛选

python 复制代码
scores = {"张三": 85, "李四": 90, "王五": 58, "赵六": 76}
passed = {name: score for name, score in scores.items() if score >= 60}
print(passed)

输出:

arduino 复制代码
{'张三': 85, '李四': 90, '赵六': 76}

✅ 字典推导式在数据过滤、重组时非常实用。


四、集合推导式(补充)

集合推导式语法与列表推导式类似,只需用 {} 包裹:

python 复制代码
unique_squares = {x * x for x in [1, 2, 2, 3, 3, 4]}
print(unique_squares)

输出:

复制代码
{16, 1, 4, 9}

集合自动去重,非常适合提取唯一值。


五、推导式的性能优势

推导式不仅让代码更简洁,还能在大多数场景下执行更快。 因为它们由 Python 内部进行了优化,减少了函数调用的开销。

python 复制代码
import time

# 普通循环
start = time.time()
result = []
for i in range(1000000):
    result.append(i * 2)
print("普通循环耗时:", time.time() - start)

# 推导式
start = time.time()
result = [i * 2 for i in range(1000000)]
print("推导式耗时:", time.time() - start)

通常情况下,推导式的执行速度更快。


六、推导式使用注意事项

  1. 不要滥用: 过于复杂的嵌套推导式会让代码难以阅读。 例如三层以上循环就应考虑使用普通循环结构。

  2. 注意内存占用 : 列表推导式会一次性生成所有数据; 如果数据量很大,可改用生成器表达式 (用圆括号 ())。


七、小结

推导式类型 语法形式 作用
列表推导式 [表达式 for 变量 in 可迭代对象 if 条件] 快速生成列表
字典推导式 {键: 值 for 变量 in 可迭代对象 if 条件] 快速生成字典
集合推导式 {表达式 for 变量 in 可迭代对象} 自动去重
生成器表达式 (表达式 for 变量 in 可迭代对象) 延迟计算,节省内存

✅ 一句话总结

推导式让 Python 代码更"像 Python"------简洁、优雅又高效。

相关推荐
田里的水稻5 分钟前
PyTorch_Dataloader\TensorBoard---数据加载与可视化
人工智能·pytorch·python·神经网络·机器学习
Dmz丶10 分钟前
前端转 AI 100 天 Day 27:Docker 容器化——从“我电脑能跑“到“到处能跑“
python
雾行0x0012 分钟前
粗排:如何在毫秒级筛出值得精排的内容?
后端·架构
卿卿的产品经理日记12 分钟前
【AI产品经理实战】Day 38|try-except 踩坑实录:报错是数据
人工智能·笔记·python·学习·产品经理
D1213813 分钟前
从零用 C++17 写一个 API 网关:线程模型、控制面分离,以及三个真实 Bug
后端
Scene21613 分钟前
AgentScope Java 进阶:2.如何精准定义问题类型与处理“不确定性”?
后端
用户3459668152113 分钟前
在嵌入式 Linux 摄像头上跑 WebRTC:metaRTC 8.0 的 8 个坑与解法
后端
今天早睡16 分钟前
用 State、Reducer 和条件边搭出可控循环
后端
站大爷IP17 分钟前
Python的字符串格式化把我坑惨了,原来f-string和format的区别这么大
后端
雾行0x0020 分钟前
精排:推荐系统如何给内容打分?从多目标预估到分数融合
后端·架构