python - 是否可以在不编写生成器函数的情况下从链表生成可迭代对象？

Question

在我正在编写的 Python 程序中，我使用字典构建了一个链表，该字典将每个节点映射到它的后继节点(最后一个节点映射到 None)。

(实际上，字典包含维基百科告诉我的内容称为 spaghetti stack ，这是一棵树，其中每个节点都链接到其父节点而不是其子节点。这意味着从叶节点开始有许多部分重叠的路径到根节点。我只关心其中一条路径，从一个特定的叶节点开始。这对问题来说并不重要，除了排除任何涉及迭代字典中所有元素的解决方案。)

我需要将此列表作为可迭代对象传递给另一个函数。我知道我可以使用生成器函数来做到这一点(请参阅下面的代码)，但似乎应该有一个内置函数来在一行中生成我需要的迭代器(或者可能是一个生成器表达式)。我对文档进行了一些搜索，但 itertools 或 functools 模块中似乎没有任何适用的内容，我不确定还能去哪里查看。

这是我现在拥有的生成器函数。可以消除(内联)外部函数，但内部生成器似乎是为数据创建可迭代对象的唯一简单方法:

def makeListGenerator(nextDict, start):
    def gen(node):
        while node:
            yield node
            node = nextDict[node]
    return gen(start)

似乎这种生成器应该有一个模式，但我不确定它会被称为什么。这是一个通用版本:

def makeGenericGenerator(nextFunc, continueFunc, start):
    def gen(value):
         while continueFunc(value):
             yield value
             value = nextFunc(value)
    return gen(start)

我可以使用这个调用实现特定版本:

makeGenericGenerator(lambda v: nextDict[v], bool, start)

Python 标准库中是否已经存在类似的东西？

Answer 1

您面临的基本问题是，每次从您的可迭代对象中获取另一个值时，您的可迭代对象都必须记住该值，以便它知道如何生成下一个值。换句话说，您的可迭代对象需要维护自己的状态。

这意味着，无论您的问题是否有好的答案，使用生成器都可能是正确的解决方案——因为这正是生成器的用途!生成器的全部意义在于它们在调用它们的 next 方法之间保存它们的状态，而这正是您所需要的。

另一方面，生成器表达式更适合无状态转换。很多人试图将状态塞进其中，这通常有点难看。实际上，我在尝试为您的案例做这件事时花了一点时间，但无法使生成器表达式起作用。我终于找到了一些有点有用的东西。这使用了鲜为人知的 callable_iterator 版本的 iter:

>>> d = {1:2, 2:3, 3:4, 4:5, 5:None}
>>> list(iter(lambda st=[1]: st.__setitem__(0, d[st[0]]) or st[0], None))
[2, 3, 4, 5]

我提出这个问题不是作为答案，而是为了说明为什么这不是很好。