2

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

（实际上，字典包含 Wikipedia 告诉我的称为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标准库中是否已经存在类似的东西？

4

2 回答 2

7

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

这意味着，无论您的问题是否有一个好的答案，使用生成器可能是正确的解决方案——因为这正是生成器的用途！生成器的全部意义在于它们在调用它们的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]

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

于 2012-06-02T22:28:07.577 回答
2 
nodes = {
    'A':'B',
    'B':'C',
    'C':'D',
    'D':None
}

print (lambda f: lambda g: f(f,g)) (lambda f,x: [x] + f(f, nodes[x]) if x else [])('A')
# ['A', 'B', 'C', 'D']
于 2012-06-02T22:55:30.127 回答