itertools
模块的文档:
https://docs.python.org/zh-cn/3/library/itertools.html
,虽然那个模块里的函数是使用 C 语言实现的,不过文档展示了如何使用 Python 实 现部分函数,这通常要利用模块里的其他函数。用法示例也很好。其中:https://docs.python.org/zh-cn/3/library/itertools.html#itertools-recipes
,说明如何使用 itertools
模块中的现有函数实现额外的高性能函数。
对于yield 讨论:
看下面这个例子:
def f():
def do_yield(n):
yield n
x = 0
while True:
x += 1
do_yield(x)
如果调用示例 中的 f()
,会得到一个无限循环,而不是生成器,因为 yield
关键 字只能把最近的外层函数变成生成器函数。虽然生成器函数看起来像函数,可是我们不能通过简单的函数调用把职责委托给另一个生成器函数。
Python 新引入的 yield from
句法允许生成器或协程把工作委托给第三方完成,这样 就无需嵌套 for
循环作为变通了。在函数调用前面加上 yield from
能“解决”示例 的问题:
def f():
def do_yield(n):
yield n
x = 0
while True:
x += 1
yield from do_yield(x)