Python中的yield关键字——生成器函数的用法

liftword3周前 (01-16)技术文章20

yield 关键字是 Python 中一个特殊关键字,用于生成器函数(generator function)中。生成器函数与普通函数的区别在于,生成器函数在遇到 yield 关键字时会暂停执行,并返回一个值。然后,当再次调用生成器函数时,会从 yield 关键字所在的位置继续执行。

生成器函数的用法

生成器函数的定义与普通函数类似,只是需要使用 yield 关键字来生成值。例如,以下代码定义了一个生成器函数,用于生成斐波那契数列:

def fib(n):
  a, b = 0, 1
  while n > 0:
    yield a
    a, b = b, a + b
    n -= 1

调用生成器函数会返回一个生成器对象。生成器对象可以迭代,每次调用 next() 方法都会返回生成器函数生成的下一个值。例如:

generator = fib(5)
print(next(generator))  # 输出 0
print(next(generator))  # 输出 1
print(next(generator))  # 输出 1
print(next(generator))  # 输出 2
print(next(generator))  # 输出 3

yield 的用法

yield 关键字可以用于生成器函数中返回多个值。例如,以下代码定义了一个生成器函数,用于生成 1 到 10 的平方数:

Python

def squares(n):
  for i in range(1, n + 1):
    yield i * i

调用生成器函数会返回一个生成器对象。生成器对象可以迭代,每次调用 next() 方法都会返回生成器函数生成的下一个平方数。例如:

Python

generator = squares(10)
print(next(generator))  # 输出 1
print(next(generator))  # 输出 4
print(next(generator))  # 输出 9
print(next(generator))  # 输出 16
print(next(generator))  # 输出 25
print(next(generator))  # 输出 36
print(next(generator))  # 输出 49
print(next(generator))  # 输出 64
print(next(generator))  # 输出 81
print(next(generator))  # 输出 100

yield 的注意事项

  • yield 关键字只能用于生成器函数中。
  • 遇到 yield 关键字时,生成器函数会暂停执行,并返回一个值。
  • 再次调用生成器函数时,会从 yield 关键字所在的位置继续执行。
  • 生成器函数可以生成多个值。
  • 生成器函数在生成完所有值后会抛出 StopIteration 异常。

yield 的应用场景

yield 关键字可以用于各种需要生成数据的场景,例如:

  • 生成斐波那契数列、杨辉三角等数学序列。
  • 生成无限序列,例如素数序列、二进制小数序列等。
  • 生成文件内容,例如逐行读取文件内容。
  • 实现协程,例如异步编程。

相关文章

少儿Python每日一题(22):杨辉三角

原题解答本次的题目如下所示:杨辉三角形又称Pascal三角形,它的第i+1行是的展开式的系数。它的一个重要性质是:三角形中的每个数字等于它两肩上的数字相加。下面给出了杨辉三角形的前4行:11 11 2...

使用 Python 的生成器(Generator)来实现生成杨辉三角

当使用 Python 的生成器来实现生成杨辉三角时,下面是一个基于你的代码的简单示例,演示如何使用生成器来生成杨辉三角的每一行:def generate_pascals_triangle():...

Python 杨氏矩形查找:高效编程新利器

#Python基础##python##python编程#一、杨氏矩形查找概述杨氏矩形查找主要是在一个二维数组中进行特定数值的查找操作。在一个每行从左到右递增、每列从上到下递增的二维数组中,通过特...

Python地铁学习计划——迭代器和生成器

地铁上不要再看“钢鞭的故事”了,关注我一起学习!前言这篇内容挺多的,而且比内容不好理解。或许新手看完后,还会一脸懵逼,不过这是正常的,如果你看完后,是迷糊的,那么建议你继续学习后面的内容,等学完,再回...

过瘾!100道Python入门练习题

实例001:数字组合实例002:“个税计算”实例003:完全平方数实例004:这天第几天实例005:三数排序实例006:斐波那契数列实例007:copy实例008:九九乘法表实例009:暂停一秒输出实...

组合数怎么计算?

组合数是指从 n 个不同元素中,任取 m (m≤n) 个元素并成一组,叫作从 n 个不同元素中取出 m 个元素的一个组合;从 n 个不同元素中取出 m (m≤n) 个元素的所有组合的个数,叫作 n 个...