递归生成器和 send()

Question

有谁知道 send() 递归使用时如何与生成器一起使用？我预计该值将传递给当前生成器，然后当前生成器可以将其传递给递归生成器......但似乎情况并非如此？一些示例代码：

def Walk(obj):
  recurse = (yield obj)
  if not recurse:
    print 'stop recurse:', recurse
    return

  if isinstance(obj, list):
    print 'is list:', obj
    for item in obj:
      print 'item loop:', item
      walker = Walk(item)
      for x in walker:
        print 'item walk:', x
        recurse = (yield x)
        print 'item walk recurse:', recurse
        walker.send(recurse)

root = ['a', ['b.0', ['b.0.0']]]

walker = Walk(root)
for i, x in enumerate(walker):
  print i, x
  print 'send true'
  walker.send(True)

所需的输出应该是每个级别递归的每个值：

0 ['a', ['b.0', ['b.0.0']]]
1 'a'
2 ['b.0', ['b.0.0']]
3 'b.0'
4 ['b.0.0']
5 'b.0.0'

最终发生的情况是：

0 ['a', ['b.0', ['b.0.0']]]
send true
is list: ['a', ['b.0', ['b.0.0']]]
item loop: a
item walk: a
item walk recurse: None
stop recurse: None

看起来带有 recurse = (yield) 的内循环不等待值待发送。或者什么。目前尚不清楚内循环递归值如何变得无；它的调用者确实调用了send()。

最终，目标基本上是递归地遍历树结构，但让最顶层的调用者能够指定何时不递归到子结构。例如，

walker = Walk(root)
for node in walker:
  if CriteriaMet(node):
    walker.send(True)
  else:
    walker.send(False)

Answer 1

需要意识到的重要一点是 send() 也会消耗！
来自 http://docs.python.org/reference/expressions.html#generator。发送：

恢复执行并将一个值“发送”到生成器函数中。
value 参数成为当前yield 表达式的结果。
send() 方法返回生成器生成的下一个值，或者
如果生成器退出而没有产生另一个生成器，则引发 StopIteration
价值。当调用send()来启动生成器时，必须调用它
以 None 作为参数，因为没有yield 表达式
可以收到值。

以下是对代码的快速修改，使其按预期输出：

def Walk(obj):
  recurse = (yield obj)
  if not recurse:
    #print 'stop recurse:', recurse
    return

  if isinstance(obj, list):
    #print 'is list:', obj
    for item in obj:
      #print 'item loop:', item
      walker = Walk(item)

      recurse = None #first send must be None
      while True:
        try:
          x = walker.send(recurse)
        except StopIteration:
          break
        #print 'item walk:', x
        recurse = (yield x)
        #print 'item walk recurse:', recurse

root = ['a', ['b.0', ['b.0.0']]]

walker = Walk(root)
i = 0
x = walker.next()
while True:
  print i, x
  try:
    x = walker.send(True)
  except StopIteration:
    break
  i += 1

输出：

0 ['a', ['b.0', ['b.0.0']]]
1 a
2 ['b.0', ['b.0.0']]
3 b.0
4 ['b.0.0']
5 b.0.0