收益率与回报的不同结果

Question

我真的不明白 yield 语句在这种情况下是如何工作的。问题是，给定一个不带括号的表达式，编写一个函数来生成所有可能的全括号 (FP) 表达式。比如说，输入是 '1+2+3+4' ，它应该生成 5 个 FP 表达式：

1. (1+(2+(3+4)))
2. (1+((2+ 3)+4))
3. ((1+2)+(3+4))
4. ((1+(2+3))+4)
5. (((1+2)+3)+4)

我的代码如下。

OPS = ('+', '-', '*', '/')
def f(expr):
    """
    Generates FP exprs
    Recursive formula: f(expr1[op]expr2) = (f(expr1) [op] f(expr2))
    """
    if expr.isdigit(): yield expr
#       return [expr]

#   ret = []
    first = ''
    i = 0
    while i < len(expr):
        if expr[i] not in OPS:
            first += expr[i]
            i += 1
        else:
            op = expr[i]
            i += 1
            second = expr[i:]
            firstG, secondG = f(first), f(second)
            for e in ('(' + e1 + op + e2 + ')' for e1 in firstG for e2 in secondG):
                yield e
#               ret.append(e)
            first += op
#    return ret

如果我使用 return 语句（注释掉的行），那么代码将按预期工作。但是，当我按照代码所示更改为 yield 语句时，我只得到前 4 个结果。如果增加输入表达式的操作数数量，那么当然会丢失更多结果。例如，对于输入 '1+2+3+4+5'，我只得到 8 而不是 14。

我最终通过注释掉该行 < 找到了使代码工作的方法code>firstG, secondaryG = f(first), f(second) 并替换行

for e in ('(' + e1 + op + e2 + ')' for e1 in firstG for e2 in secondaryG):

by

for e in ('(' + e1 + op + e2 + ')' for e1 in f(first) for e2 in f(second)):

这意味着由于 firstG, secondaryG = f(first), f(second) 行，生成器的一些“信息”丢失了，但我无法弄清楚真正的原因。你们能给我一些想法吗？

Answer 1

问题是您正在迭代生成器而不是yield版本中的列表，特别是在一次循环后耗尽的secondG。将行更改为此，它就可以工作：

firstG, secondG = f(first), list(f(second))

或者，您可以更改循环：

for e in ("(%s%s%s)" % (e1, op, e2) for e1 in f(first) for e2 in f(second)):
#                               new generator object every loop  ^^^^^^^^^

非yield版本可以工作，因为您返回列表，与生成器不同，列表可以再次迭代。另请注意，您仅迭代 firstG 一次，因此它不受影响。

请记住：

r = [v for a in A for b in B]

相当于：

r = []
for a in A:
  for b in B:
    r.append(v)

这更清楚地显示了 B 上的重复循环。

另一个例子：

def y():
  yield 1
  yield 2
  yield 3
def r():
  return [1, 2, 3]

vy = y()
for v in vy:
  print v
for v in vy:
  print v

print "---"

vr = r()
for v in vr:
  print v
for v in vr:
  print v