Python中的yield问题|使用辅助函数

Question

给定一个包含很多单词的字符串，我想颠倒单词的顺序。如果输入为Thanks for all the fish，则输出应为fish the all for Thanks。

我正在尝试使用 generators 解决此问题，以避免创建新列表。 我想出了以下解决方案：

from itertools import islice

def foo(s, begin, end):
    for elem in islice(s, begin, end):
        yield elem

def baz(s):
    i = len(s)
    j = len(s)

    while True:
        i -= 1
        if i == -1:
            foo(s, i+1, j)
            break
        if s[i] == ' ':
            foo(s, i+1, j)
            yield ' '
            j = i


s = "Thanks for all the fish"
g = baz(s)
for x in g: print(x, end='')

然而，这段代码的输出是" "（一个只有空格的字符串）。

另一方面，如果我直接print 元素而不是产生它们，它会起作用：

from itertools import islice

def foo(s, begin, end):
    for elem in islice(s, begin, end):
        print(elem, end='')

def baz(s):
    i = len(s)
    j = len(s)

    while True:
        i -= 1
        if i == -1:
            foo(s, i+1, j)
            break
        if s[i] == ' ':
            foo(s, i+1, j)
            print(' ', end='')
            j = i


s = "Thanks for all the fish"
g = baz(s)

这有效，并作为输出fish the all for Thanks。但我不想仅仅能够打印它，我想要一个正确的生成器。

最后，我发现如果避免调用foo 它也可以：

from itertools import islice

def baz(s):
    i = len(s)
    j = len(s)

    while True:
        i -= 1
        if i == -1:
            for elem in islice(s, i+1, j):
                yield elem
            break
        if s[i] == ' ':
            for elem in islice(s, i+1, j):
                yield elem
            yield ' '
            j = i


s = "Thanks for all the fish"
g = baz(s)
for x in g: print(x, end='')

这个的输出是fish the all for Thanks。 这个版本给了我我想要的东西（一个正确的生成器），但我在它的函数中重复了代码。

我在堆栈交换线程中提到了 yield 的作用，但我可能理解错了。 我的理解是生成器将一直运行，直到找到下一个 yield 语句（或函数结束）。 这个问题让我觉得这不是它的工作原理，但除此之外我一无所知。

如何使用第一个代码 sn-p 中的帮助函数来解决这个问题？

Answer 1

您可以使用yield from foo(...)（另请参阅PEP-380 on "Syntax for Delegating to a Subgenerator"）或for whatever in foo(...): yield whatever 来连续yield 生成器中的元素：

def baz(s):
    i = len(s)
    j = len(s)

    while True:
        i -= 1
        if i == -1:
            yield from foo(s, i+1, j)    # yield from syntax
            break
        if s[i] == ' ':
            for item in foo(s, i+1, j):  # for-loop over generator
                yield item
            yield ' '
            j = i

但是，正如@trentcl 在 cmets 中指出的那样，在您的情况下，您不需要 foo-helper 函数，因为它基本上也完成了 islice 的工作。所以你可以简单地将foo 替换为islice：

def baz(s):
    i = len(s)
    j = len(s)

    while True:
        i -= 1
        if i == -1:
            yield from islice(s, i+1, j)    # yield from syntax
            break
        if s[i] == ' ':
            for item in islice(s, i+1, j):  # for-loop over generator
                yield item
            yield ' '
            j = i