python:yield用法

yield 是 Python 中用于定义**生成器(generator)**的关键字。与 return 不同,yield 在函数中每次调用时会"产出"一个值,并暂停函数的执行状态;下次再调用时,从上次暂停的位置继续执行。


一、基本用法

示例 1:简单生成器

复制代码
def simple_generator():
    yield 1
    yield 2
    yield 3

gen = simple_generator()
print(next(gen))  # 输出: 1
print(next(gen))  # 输出: 2
print(next(gen))  # 输出: 3
# print(next(gen))  # 抛出 StopIteration 异常

也可以用 for 循环遍历:

复制代码
for value in simple_generator():
    print(value)
# 输出:
# 1
# 2
# 3

二、yield vs return

  • return:函数执行完就结束,返回一个值。
  • yield:函数变成生成器,可以多次返回值,每次调用恢复执行。

三、带参数的生成器(生成器表达式)

复制代码
def countdown(n):
    while n > 0:
        yield n
        n -= 1

for i in countdown(5):
    print(i)
# 输出: 5 4 3 2 1

这种写法非常节省内存,因为不会一次性生成所有值。

相关推荐
dapeng287012 分钟前
分布式系统容错设计
开发语言·c++·算法
2501_9454235415 分钟前
用Matplotlib绘制专业图表:从基础到高级
jvm·数据库·python
2301_7938046916 分钟前
使用Python处理计算机图形学(PIL/Pillow)
jvm·数据库·python
qq_4176950517 分钟前
代码热修复技术
开发语言·c++·算法
badhope5 小时前
Mobile-Skills:移动端技能可视化的创新实践
开发语言·人工智能·git·智能手机·github
码云数智-园园6 小时前
微服务架构下的分布式事务:在一致性与可用性之间寻找平衡
开发语言
吴佳浩6 小时前
GPU 编号进阶:CUDA\_VISIBLE\_DEVICES、多进程与容器化陷阱
人工智能·pytorch·python
C++ 老炮儿的技术栈7 小时前
volatile使用场景
linux·服务器·c语言·开发语言·c++
hz_zhangrl7 小时前
CCF-GESP 等级考试 2026年3月认证C++一级真题解析
开发语言·c++·gesp·gesp2026年3月·gespc++一级
Liu628887 小时前
C++中的工厂模式高级应用
开发语言·c++·算法