Python 列表推导式:让你的代码更优雅、更高效

一、一个平平无奇的 for 循环

想象一个简单的任务:我们有一个数字列表 [1, 2, 3, 4, 5],现在想得到一个新列表,其中每个数字都是原列表数字的平方。

按照我们最熟悉的思路,你会这样写:

python 复制代码
numbers = [1, 2, 3, 4, 5]
squared_numbers = []  # 1. 先创建一个空列表

for num in numbers:  # 2. 遍历原列表
    squared_numbers.append(num * num)  # 3. 计算平方并添加进去

print(squared_numbers)
# 输出: [1, 4, 9, 16, 25]

二、列表推导式的真面目

同样的需求,用列表推导式来写,只需要一行:

python 复制代码
numbers = [1, 2, 3, 4, 5]
squared_numbers_new = [num * num for num in numbers]

print(squared_numbers_new)
# 输出: [1, 4, 9, 16, 25]

解析

[<表达式> for <元素> in <可迭代对象>]

  • []:首先,它被一对方括号包围,这明确告诉Python:"嘿,我要创建一个新列表!"
  • <表达式> (num * num):这是你希望对每个元素进行的操作。可以是计算、调用函数,或者任何能返回一个值的表达式。
  • for <元素> in <可迭代对象> (for num in numbers):这部分和你熟悉的 for 循环几乎一模一样,它负责从源(numbers列表)中逐个取出元素(num)。

三、加入if筛选

列表推导式的强大之处不止于此。假设我们只想计算偶数的平方,怎么办?

传统的 for 循环需要加一个 if 判断:

python 复制代码
numbers = [1, 2, 3, 4, 5]
even_squared_numbers = []

for num in numbers:
    if num % 2 == 0:  # 如果是偶数
        even_squared_numbers.append(num * num)

print(even_squared_numbers)
# 输出: [4, 16]

而在列表推导式中,我们只需要在末尾加上 if 条件即可:

python 复制代码
numbers = [1, 2, 3, 4, 5]
even_squared_numbers = [num * num for num in numbers if num % 2 == 0]

print(even_squared_numbers)
# 输出: [4, 16]

语法变成了:[<表达式> for <元素> in <可迭代对象> if <条件>]

这个 if 就像一个过滤器,只有满足条件的元素才会被加工并放入新列表。

四、更进一步的if-else

那如果我想对不同条件的元素做不同的处理呢?比如,把偶数变成它的平方,奇数变成它的相反数。

这时,if-else就要登场了。但要注意,它的位置和单纯的 if 筛选不一样。

python 复制代码
numbers = [1, 2, 3, 4, 5]
transformed_numbers = [num * num if num % 2 == 0 else -num for num in numbers]

print(transformed_numbers)
# 输出: [-1, 4, -3, 16, -5]

它的语法是:[<为真时的表达式> if <条件> else <为假时的表达式> for <元素> in <可迭代对象>]

这里的 if-else 构成了一个三元运算符 ,它必须放在 for 循环的前面,因为它决定了表达式本身的值。

结语

点个赞,关注我获取更多实用 Python 技术干货!如果觉得有用,记得收藏本文!

相关推荐
killerbasd1 分钟前
还是迷茫 5.3
前端·react.js·前端框架
其实防守也摸鱼6 分钟前
CTF密码学综合教学指南--第五章
开发语言·网络·笔记·python·安全·网络安全·密码学
不会敲代码11 小时前
TCP/IP 与前端性能:从数据包到首次渲染的底层逻辑
前端·tcp/ip
kyriewen1 小时前
奥特曼借GPT-5.5干杯,而你的Copilot正按Token收钱
前端·github·openai
callJJ1 小时前
Spring Data Redis 两种编程模型详解:同步 vs 响应式
java·spring boot·redis·python·spring
小郑加油1 小时前
python学习Day12:pandas安装与实际运用
开发语言·python·学习
AC赳赳老秦1 小时前
投标合规提效:用 OpenClaw 实现标书 / 合同自动审核、关键词校验、格式优化,降低废标风险
开发语言·前端·python·eclipse·emacs·deepseek·openclaw
kyriewen1 小时前
代码写成一锅粥?3个设计模式让你的项目“起死回生”
前端·javascript·设计模式
.柒宇.1 小时前
AI掘金头条项目-K8s部署实战教程
python·云原生·容器·kubernetes·fastapi
观北海2 小时前
从 Sim2Sim 到 Sim2Real:以 ONNX 为核心的机器人策略实机落地全指南
python·机器人