这种"一行搞定"的风格统称为 Comprehensions(推导式) ,是 Python 最具特色的语法之一。 我们从列表推导式说起~
一、从"普通方式"创建列表说起
python
# 传统方式
squares = []
for i in range(1, 6):
squares.append(i ** 2)
print(squares) # 输出: [1, 4, 9, 16, 25]
# 列表推导式
squares = [i ** 2 for i in range(1, 6)]
print(squares) # 输出: [1, 4, 9, 16, 25]
二、拆解列表推导式
列表推导式是一种用更简洁、更 Pythonic 的方式创建列表的语法。(其实就是把传统方式换个写法)
对于上面所用到的列表推导式:
python
# 列表推导式
squares = [i ** 2 for i in range(1, 6)]
print(squares) # 输出: [1, 4, 9, 16, 25]
拆解一下这行代码:
i ** 2
:这是"表达式",表示我们最终要放进列表里的东西(每个元素的值)。for i in range(1, 6)
:这是循环部分,i
是临时变量,遍历range(1, 6)
中的每个值。- 整个用方括号
[]
包起来,表示结果是一个列表。
✅记住:列表推导式的结果一定是一个列表!
三、加上条件判断if
和else
python
# 过滤条件:不要奇数
even_squares = [i ** 2 for i in range(1, 6) if i % 2 == 0]
print(even_squares) # 输出: [4, 16]
# 规则:偶数平方,奇数变成负数
results = [i ** 2 if i % 2 == 0 else -i for i in range(1, 6)]
print(results) # 输出: [-1, 4, -3, 16, -5]
列表推导式的基本结构本来长这样:
python
[表达式 for 临时变量 in 可迭代对象 if 条件]
if
放在最后,用来过滤元素。- 只有满足
if
条件的i
,才会被计算表达式
并加入列表。
但是,如果要对每个元素做不同处理 (比如偶数平方,奇数变负数),就不能把 if
放在最后了。
结构变成:
python
[表达式1 if 条件 else 表达式2 for 临时变量 in 可迭代对象]
⚠️ 注意区别:
if
在 最后 :用于过滤(不满足条件的直接跳过)if-else
在 表达式位置 :用于对每个元素做不同处理(所有元素都会被处理)
四、嵌套循环(不要过度嵌套,会很难读)
python
colors = ['红', '蓝']
sizes = ['S', 'M']
# 传统写法:
pairs = []
for c in colors:
for s in sizes:
pairs.append((c, s))
# 列表推导式写法:
pairs = [(c, s) for c in colors for s in sizes]
print(pairs)
# 输出: [('红', 'S'), ('红', 'M'), ('蓝', 'S'), ('蓝', 'M')]
五、一些拓展
5.1 其它推导式

5.2 生成器表达式
列表推导式会一次性生成整个列表 ,如果数据很大,可能占用很多内存。这时可以考虑用生成器表达式 (用圆括号 ()
而不是 []
)
python
gen = (i**2 for i in range(1, 6))
print(gen) # <generator object ...>
# 用 for 循环取值:
for x in gen:
print(x) # 1, 4, 9, 16, 25
💡 重要区别:
[...]
:立刻创建整个列表,占内存。(...)
:只创建一个"生成器",用多少算多少,省内存,适合大数据。
六、总结一句话
列表推导式 = 用一行代码,把"循环 + 条件 + 表达式"打包成一个新列表。