Python 教程

Python 教程 Python 简介 Python 历史 Python 下载安装和环境配置 Python 入门 Python 语法 Python 输出 Python 注释 Python 变量 Python 数据类型 Python 数值类型 Python 类型转换 Python 字符串 Python 布尔值 Python 运算符 Python 列表 Python 元组 Python 集合 Python 字典 Python If...Else Python While 循环 Python For 循环 Python 函数 Python Range Python 数组 Python 类和对象 Python 继承 Python 迭代 Python 模块 Python 日期时间 Python 数学运算 Python JSON Python 正则表达式 Python PIP Python Try...Except Python 用户输入 Python 字符串格式化

Python 文件处理

Python 文件处理 Python 打开文件 Python 创建/写入文件 Python 删除文件

Python NumPy

NumPy 简介 NumPy 入门 NumPy 创建数组 NumPy 数组索引 NumPy 数组裁切 NumPy 数据类型 NumPy 副本 vs 视图 NumPy 数组形状 NumPy 数组重塑 NumPy 数组迭代 NumPy 数组连接 NumPy 数组拆分 NumPy 数组搜索 NumPy 数组排序 NumPy 数组过滤 NumPy 随机数 NumPy ufunc 通用函数

Python SciPy

SciPy 简介 SciPy 入门 SciPy 常量 SciPy 优化器 SciPy 稀疏数据 SciPy 图表 SciPy 空间数据 SciPy Matlab 数组 SciPy 插值 SciPy 统计显着性检验

Python 机器学习

Machine 机器学习入门 Machine 平均中位数模式 Machine 标准差 Machine 百分位数 Machine 数据分布 Machine 正态数据分布 Machine 散点图 Machine 线性回归 Machine 多项式回归 Machine 多元回归 Machine 缩放 Machine 训练/测试 Machine 决策树

Python MySQL

MySQL 入门 MySQL Create Database MySQL Create Table MySQL Insert MySQL Select MySQL Where MySQL Order By MySQL Delete MySQL Drop Table MySQL Update MySQL Limit MySQL Join

Python MongoDB

MongoDB 入门 MongoDB 创建数据库 MongoDB 创建集合 MongoDB 插入 MongoDB 查找 MongoDB 查询 MongoDB 排序 MongoDB 删除 MongoDB 删除集合 MongoDB 更新 MongoDB 限制

Python 参考手册

Python 参考手册 Python 内置函数 Python 字符串方法 Python 列表/数组方法 Python 字典方法 Python 元组方法 Python 集合方法 Python 文件方法 Python 关键字 Python 内置异常 Python 词汇表

Python 模块参考

Python 内置模块 Python random 随机模块 Python requests 请求模块 Python statistics 统计模块 Python math 数学模块 Python cMath 模块

Python 如何使用

Python 删除列表重复项 Python 反转字符串 Python 添加两个数字

Python 高级教程

Python 常用指引 将Python2代码迁移到Python3 将扩展模块移植到 Python3 Curses 编程 描述器使用指南 函数式编程指引 日志常用指引 日志操作手册 正则表达式使用指南 套接字编程指南 排序指南 Unicode 指南 如何利用urllib包获取网络资源 Argparse 教程 ipaddress 模块介绍 Argument Clinic 的用法 使用DTrace和SystemTap检测CPython 对象注解属性的最佳实践

Python 实例

Python 实例 Python 编译器 Python 练习 Python 测验 NumPy 测验 SciPy 测验


Python 生成器

生成器

生成器是可以暂停和恢复执行的函数。

调用生成器函数时,它会返回一个生成器对象,也就是一个迭代器。

函数内部的代码尚未执行,仅被编译。只有当你遍历生成器时,函数才会执行。

实例

一个简单的生成器函数:

def my_generator():
  yield 1
  yield 2
  yield 3

for value in my_generator():
  print(value)
亲自试一试 »

生成器允许你遍历数据,而无需将整个数据集存储在内存中。

生成器不使用 return,而是使用 yield 关键字。


yield 关键字

yield 关键字使函数成为生成器。

当遇到 yield 关键字时,函数的状态会被保存,并返回该值。下次调用生成器时,它会从上次中断的地方继续执行。

实例

生成数字的生成器:

def count_up_to(n):
  count = 1
  while count <= n:
    yield count
    count += 1

for num in count_up_to(5):
  print(num)
亲自试一试 »

与会终止函数的 return 不同,yield 会暂停函数,并且可以多次调用。



生成器节省内存

生成器之所以节省内存,是因为它们可以动态生成值,而不是将所有数据都存储在内存中。

对于大型数据集,生成器可以节省内存:

实例

大型序列生成器:

def large_sequence(n):
  for i in range(n):
    yield i

# 这不会在内存中创建一百万个数字
gen = large_sequence(1000000)
print(next(gen))
print(next(gen))
print(next(gen))
亲自试一试 »

使用 next() 函数处理生成器

您可以使用 next() 函数手动遍历生成器:

实例

def simple_gen():
  yield "Emil"
  yield "Tobias"
  yield "Linus"

gen = simple_gen()
print(next(gen))
print(next(gen))
print(next(gen))
亲自试一试 »

当没有更多值可供生成时,生成器会引发一个StopIteration异常:

实例

def simple_gen():
  yield 1
  yield 2

gen = simple_gen()
print(next(gen))
print(next(gen))
print(next(gen)) # 这将引发 StopIteration 异常
亲自试一试 »

生成器表达式

类似于列表推导式,您可以使用圆括号而不是方括号来创建生成器:

实例

列表推导式与生成器表达式:

# 列表推导式 - 创建一个列表
list_comp = [x * x for x in range(5)]
print(list_comp)

# 生成器表达式 - 创建一个生成器
gen_exp = (x * x for x in range(5))
print(gen_exp)
print(list(gen_exp))
亲自试一试 »

实例

使用生成器表达式求和:

# 无需创建列表即可计算平方和
total = sum(x * x for x in range(10))
print(total)
亲自试一试 »

斐波那契数列生成器

生成器可以用来生成斐波那契数列。

它可以无限期地生成数值,而不会耗尽内存:

实例

生成 100 个斐波那契数:

def fibonacci():
  a, b = 0, 1
  while True:
    yield a
    a, b = b, a + b

获取前100个斐波那契数列数字
gen = fibonacci()
for _ in range(100):
  print(next(gen))
亲自试一试 »

生成器方法

生成器具有用于高级控制的特殊方法:

send() 方法

send() 方法允许您向生成器发送一个值:

实例

def echo_generator():
  while True:
    received = yield
    print("Received:", received)

gen = echo_generator()
next(gen) # Prime the generator
gen.send("Hello")
gen.send("World")
亲自试一试 »

close() 方法

close() 方法会停止生成器:

实例

def my_gen():
  try:
    yield 1
    yield 2
    yield 3
  finally:
    print("Generator closed")

gen = my_gen()
print(next(gen))
gen.close()
亲自试一试 »