return 语句中的 for 循环

Question

我正在研究来自 What is the most "pythonic" way to iterate over a list in chunks? 的 Nosklo 的回复，他在那里定义了函数：

def chunker(seq, size):
    return (seq[pos:pos + size] for pos in xrange(0, len(seq), size))

有人可以解释一下在 for 循环之后返回是如何工作的吗？ 我尝试执行以下操作：

def chunker2(seq, size):
    for pos in xrange(0, len(seq), size):
    return seq[pos:pos + size]

但我没有得到相同的结果。 请注意，在 Nosklo 示例中，chunker() 被迭代调用，如下例所示：

animals = ['cat', 'dog', 'rabbit', 'duck', 'bird', 'cow', 'gnu', 'fish']
for group in chunker(animals, 3):
    print group

通过添加打印，我注意到后者 for 循环执行了 3 次（它通过 animals 列表），但 chunker 函数中的 for 循环仅执行一次。那么，为什么只有一个返回，我可以看到 3 个打印？

Answer 1

nosklo chunker 函数的返回值称为生成器，是一个在迭代时会生成值的对象。在这种情况下，生成器是使用 generator expression 创建的，这是括号之间不可分割的一段代码：(seq[pos:pos + size] for pos in xrange(0, len(seq), size))。

>>> def chunker(seq, size):
...     return (seq[pos:pos + size] for pos in xrange(0, len(seq), size))
... 
>>> result = chunker([1, 2, 3], 2)
>>> print(result)
<generator object <genexpr> at 0x10581e1e0>
>>> next(result)
[1, 2]
>>> next(result)
[3]

关于调用什么，我们可以这样重写循环代码以更好地查看：

>>> generator = chunker(animals, 3)
>>> for chunk in generator:
...     print chunk
... 
['cat', 'dog', 'rabbit']
['duck', 'bird', 'cow']
['gnu', 'fish']    

chunker 函数只被调用一次并返回生成器对象，我们将其存储在generator 变量中。然后for 循环仅适用于这个generator 对象并调用它3 次。

为了能够打印此生成器的实际调用，您必须在生成器表达式 (seq[pos:pos + size] for pos in xrange(0, len(seq), size)) 中包含 print 语句，这是不允许的。但是我们可以使用yield 语句将这个生成器表达式重写为普通的generator function，这是一种更详细但也更通用的生成器形式，我们可以在其中包含print 语句，它会按照您最初的预期工作：

>>> def chunker2(seq, size):
...     for pos in xrange(0, len(seq), size):
...         print('chunker2 generator called')
...         yield seq[pos:pos + size]
... 
>>> for group in chunker2(animals, 3):
...     print group
... 
chunker2 generator called
['cat', 'dog', 'rabbit']
chunker2 generator called
['duck', 'bird', 'cow']
chunker2 generator called
['gnu', 'fish']

这里的 chunker2 函数本身就是真正的生成器，被调用了 3 次。