0

0

Python中生成器函数用法详解 Python中yield关键字教程

尼克

尼克

发布时间:2025-08-27 16:43:01

|

209人浏览过

|

来源于php中文网

原创

生成器函数与普通函数的本质区别在于:普通函数执行后返回值并销毁状态,而生成器函数通过yield暂停并保持状态,返回生成器对象实现惰性求值和内存高效迭代。

python中生成器函数用法详解 python中yield关键字教程

Python中的生成器函数和

yield
关键字,是处理大量数据或构建高效迭代器时非常强大的工具。它们的核心思想在于“按需生成”数据,而不是一次性将所有数据加载到内存中,这在很多场景下能显著提升性能和资源利用率。简单来说,
yield
让一个函数能够暂停执行并返回一个值,然后在下次需要时从暂停的地方继续执行。

当我第一次接触

yield
的时候,说实话,感觉有点“反直觉”。我们习惯了函数执行到
return
就彻底结束,但
yield
完全打破了这个模式。一个函数如果包含了
yield
语句,它就不再是一个普通的函数,而变成了一个生成器函数。调用它并不会立即执行函数体,而是返回一个生成器对象。这个对象才是真正可迭代的。

每次我们从生成器对象中请求下一个值(比如通过

for
循环或者手动调用
next()
),生成器函数就会从上次暂停的地方继续执行,直到遇到下一个
yield
语句,再次暂停并“吐出”一个值。当函数体执行完毕,或者遇到
return
(在生成器中
return
会终止迭代,并可选地返回一个值,但通常我们不这么用),生成器就会抛出
StopIteration
异常,表示没有更多的数据了。

这种“惰性求值”的机制,是我认为

yield
最迷人的地方。想象一下,你要处理一个几GB甚至几十GB的文件,如果一次性读入内存,你的程序可能直接就崩溃了。但用生成器,你可以一行一行地读取、处理,每次只占用极少的内存。

立即学习Python免费学习笔记(深入)”;

# 一个简单的生成器函数示例
def my_range(n):
    i = 0
    while i < n:
        yield i
        i += 1

# 使用生成器
print("通过for循环迭代:")
for num in my_range(5):
    print(num)

print("\n手动通过next()获取:")
gen = my_range(3)
print(next(gen)) # 输出 0
print(next(gen)) # 输出 1
print(next(gen)) # 输出 2
try:
    print(next(gen)) # 尝试获取第四个值,会触发StopIteration
except StopIteration:
    print("迭代结束了!")

你看,它就像一个聪明的工厂,不是一次性生产所有产品,而是接到订单才生产一个,然后等着下一个订单。

Python生成器函数与普通函数有何本质区别?

我觉得理解生成器和普通函数的核心差异,是掌握

yield
的关键一步。最直观的区别在于,普通函数执行到
return
语句后,会立即终止,并将
return
后的值(或
None
)作为结果返回给调用者。函数的局部变量和执行状态都会被销毁。

而生成器函数,它遇到

yield
后,会暂停当前的执行,将
yield
后面的值返回。但关键在于,它会“记住”当前的执行状态——包括所有的局部变量和执行到哪一行。下次调用它时,它会从上次暂停的地方继续执行,而不是从头开始。这就像你读一本小说,读到一半合上书,下次打开时能直接从上次的章节继续读,而不是每次都从第一页开始。

从技术层面讲,普通函数返回的是一个具体的值,而生成器函数返回的是一个迭代器(或者说生成器对象),这个对象实现了迭代器协议(即有

__iter__
__next__
方法)。这意味着你可以用
for
循环去遍历它,或者用
next()
函数逐个获取值。这种差异,直接决定了它们在内存管理和处理大型数据集时的效率天壤之别。

在哪些实际场景下,我们应该优先考虑使用Python生成器?

我个人在使用Python处理数据时,一旦发现需要迭代处理大量数据,或者数据源是流式的、无限的,我几乎会本能地想到生成器。

GentleAI
GentleAI

GentleAI是一个高效的AI工作平台,为普通人提供智能计算、简单易用的界面和专业技术支持。让人工智能服务每一个人。

下载
  1. 处理大型文件或数据集: 这是最典型的应用场景。比如你需要读取一个日志文件,分析其中包含特定关键字的行。如果文件有几十GB,你不可能一次性读入内存。用生成器,你可以逐行读取,逐行处理,内存占用始终保持在一个非常低的水平。

    def read_large_file(filepath):
        # 实际应用中,这里应该处理文件不存在等异常
        try:
            with open(filepath, 'r', encoding='utf-8') as f:
                for line in f:
                    yield line.strip()
        except FileNotFoundError:
            print(f"文件未找到: {filepath}")
            # 可以选择抛出异常或返回空生成器
            return
    
    # 假设有一个很大的'access.log'文件
    # 创建一个模拟的大文件
    with open('test_access.log', 'w') as f:
        for i in range(10000):
            f.write(f"Line {i}: This is a log entry. {'ERROR' if i % 100 == 0 else ''}\n")
    
    print("\n处理模拟日志文件:")
    for log_entry in read_large_file('test_access.log'):
        if "ERROR" in log_entry:
            print(f"发现错误日志: {log_entry}")
  2. 生成无限序列: 比如斐波那契数列,或者一个计数器。你无法一次性生成所有这些数字,因为它们是无限的。生成器可以按需提供。

    def fibonacci_sequence():
        a, b = 0, 1
        while True: # 无限循环,因为斐波那契数列是无限的
            yield a
            a, b = b, a + b
    
    print("\n生成斐波那契数列的前10个数字:")
    fib_gen = fibonacci_sequence()
    for _ in range(10):
        print(next(fib_gen)) # 打印前10个斐波那契数
  3. 数据管道和流式处理: 当你需要对数据进行一系列转换时,生成器可以构建一个高效的“管道”。每个生成器函数负责一个步骤,数据在管道中流动,每次只处理一小块,而不是在每个步骤都创建中间列表。

    def filter_evens(numbers):
        for num in numbers:
            if num % 2 == 0:
                yield num
    
    def square_numbers(numbers):
        for num in numbers:
            yield num * num
    
    data = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
    # 传统方式可能需要创建中间列表,占用更多内存
    # evens = [n for n in data if n % 2 == 0]
    # squared_evens = [n*n for n in evens]
    
    # 生成器管道
    print("\n通过生成器管道处理数据:")
    processed_data = square_numbers(filter_evens(data))
    for res in processed_data:
        print(res) # 输出 4, 16, 36, 64, 100

    这种链式调用,在我看来,不仅高效,代码也更清晰。

Python中的生成器表达式(Generator Expression)是什么?它与生成器函数有何异同?

当我们谈到生成器时,除了生成器函数,还有一个非常简洁且同样强大的工具——生成器表达式。我常常把它看作是列表推导式(List Comprehension)的“惰性”版本。

生成器表达式的语法非常像列表推导式,只是它使用圆括号

()
而不是方括号
[]

# 列表推导式:立即创建一个包含所有元素的列表
# my_list = [x * x for x in range(1000000)] # 会立即占用大量内存,慎用大范围

# 生成器表达式:创建一个生成器对象,按需生成值
my_gen_exp = (x * x for x in range(1000000)) # 几乎不占用内存,直到你迭代它

print("\n生成器表达式示例:")
print(f"生成器表达式对象: {my_gen_exp}")
print(f"第一个值: {next(my_gen_exp)}") # 输出 0
print(f"第二个值: {next(my_gen_exp)}") # 输出 1
# 我们可以继续迭代
for _ in range(3):
    print(next(my_gen_exp)) # 输出 4, 9, 16

区别:

  • 语法: 生成器函数使用
    def
    定义和
    yield
    关键字;生成器表达式是类似推导式的单行语法,用圆括号包裹。
  • 返回类型: 生成器函数返回一个生成器对象;生成器表达式直接生成一个生成器对象。
  • 复杂性: 生成器函数可以包含多行逻辑、条件判断、循环甚至异常处理,适合复杂的逻辑;生成器表达式更适合简单的、单行可表达的转换或过滤操作。
  • 重用性: 生成器函数可以有参数,可以被多次调用以创建新的生成器实例;生成器表达式通常是即时创建,如果需要多次迭代同一个逻辑,可能需要重新创建表达式(或者将其封装到函数中)。

相同点:

  • 惰性求值: 两者都实现了惰性求值,按需生成数据,而不是一次性生成所有数据。
  • 内存效率: 都非常内存高效,尤其是在处理大型数据集时。
  • 迭代器协议: 两者都返回实现了迭代器协议的对象,可以被
    for
    循环遍历,或通过
    next()
    获取值。

在我看来,如果你需要一个简单的、一次性的迭代器,生成器表达式是优雅且高效的选择。但如果逻辑更复杂,或者你需要封装更强的复用性,那么生成器函数无疑是更好的工具。选择哪一个,更多时候取决于你代码的复杂度和可读性需求。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
TypeScript类型系统进阶与大型前端项目实践
TypeScript类型系统进阶与大型前端项目实践

本专题围绕 TypeScript 在大型前端项目中的应用展开,深入讲解类型系统设计与工程化开发方法。内容包括泛型与高级类型、类型推断机制、声明文件编写、模块化结构设计以及代码规范管理。通过真实项目案例分析,帮助开发者构建类型安全、结构清晰、易维护的前端工程体系,提高团队协作效率与代码质量。

26

2026.03.13

Python异步编程与Asyncio高并发应用实践
Python异步编程与Asyncio高并发应用实践

本专题围绕 Python 异步编程模型展开,深入讲解 Asyncio 框架的核心原理与应用实践。内容包括事件循环机制、协程任务调度、异步 IO 处理以及并发任务管理策略。通过构建高并发网络请求与异步数据处理案例,帮助开发者掌握 Python 在高并发场景中的高效开发方法,并提升系统资源利用率与整体运行性能。

46

2026.03.12

C# ASP.NET Core微服务架构与API网关实践
C# ASP.NET Core微服务架构与API网关实践

本专题围绕 C# 在现代后端架构中的微服务实践展开,系统讲解基于 ASP.NET Core 构建可扩展服务体系的核心方法。内容涵盖服务拆分策略、RESTful API 设计、服务间通信、API 网关统一入口管理以及服务治理机制。通过真实项目案例,帮助开发者掌握构建高可用微服务系统的关键技术,提高系统的可扩展性与维护效率。

178

2026.03.11

Go高并发任务调度与Goroutine池化实践
Go高并发任务调度与Goroutine池化实践

本专题围绕 Go 语言在高并发任务处理场景中的实践展开,系统讲解 Goroutine 调度模型、Channel 通信机制以及并发控制策略。内容包括任务队列设计、Goroutine 池化管理、资源限制控制以及并发任务的性能优化方法。通过实际案例演示,帮助开发者构建稳定高效的 Go 并发任务处理系统,提高系统在高负载环境下的处理能力与稳定性。

51

2026.03.10

Kotlin Android模块化架构与组件化开发实践
Kotlin Android模块化架构与组件化开发实践

本专题围绕 Kotlin 在 Android 应用开发中的架构实践展开,重点讲解模块化设计与组件化开发的实现思路。内容包括项目模块拆分策略、公共组件封装、依赖管理优化、路由通信机制以及大型项目的工程化管理方法。通过真实项目案例分析,帮助开发者构建结构清晰、易扩展且维护成本低的 Android 应用架构体系,提升团队协作效率与项目迭代速度。

92

2026.03.09

JavaScript浏览器渲染机制与前端性能优化实践
JavaScript浏览器渲染机制与前端性能优化实践

本专题围绕 JavaScript 在浏览器中的执行与渲染机制展开,系统讲解 DOM 构建、CSSOM 解析、重排与重绘原理,以及关键渲染路径优化方法。内容涵盖事件循环机制、异步任务调度、资源加载优化、代码拆分与懒加载等性能优化策略。通过真实前端项目案例,帮助开发者理解浏览器底层工作原理,并掌握提升网页加载速度与交互体验的实用技巧。

102

2026.03.06

Rust内存安全机制与所有权模型深度实践
Rust内存安全机制与所有权模型深度实践

本专题围绕 Rust 语言核心特性展开,深入讲解所有权机制、借用规则、生命周期管理以及智能指针等关键概念。通过系统级开发案例,分析内存安全保障原理与零成本抽象优势,并结合并发场景讲解 Send 与 Sync 特性实现机制。帮助开发者真正理解 Rust 的设计哲学,掌握在高性能与安全性并重场景中的工程实践能力。

227

2026.03.05

PHP高性能API设计与Laravel服务架构实践
PHP高性能API设计与Laravel服务架构实践

本专题围绕 PHP 在现代 Web 后端开发中的高性能实践展开,重点讲解基于 Laravel 框架构建可扩展 API 服务的核心方法。内容涵盖路由与中间件机制、服务容器与依赖注入、接口版本管理、缓存策略设计以及队列异步处理方案。同时结合高并发场景,深入分析性能瓶颈定位与优化思路,帮助开发者构建稳定、高效、易维护的 PHP 后端服务体系。

532

2026.03.04

AI安装教程大全
AI安装教程大全

2026最全AI工具安装教程专题:包含各版本AI绘图、AI视频、智能办公软件的本地化部署手册。全篇零基础友好,附带最新模型下载地址、一键安装脚本及常见报错修复方案。每日更新,收藏这一篇就够了,让AI安装不再报错!

171

2026.03.04

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.9万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号