Question

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

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

我需要将此列表作为iterable传递给另一个函数。我知道我可以使用生成器函数来执行此操作（请参阅下面的代码），但似乎应该有一个内置函数来在一行中生成我需要的迭代器（或者可能是生成器表达式）。我已经对文档进行了一些搜索，但是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

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

这意味着，无论你的问题是否有一个好的答案，使用发电机可能是正确的解决方案 - 因为这正是发电机的制造商！生成器的重点在于它们在调用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]

我提出这不是一个答案，而是为了证明为什么这不能很好地解决。

Answer 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']

是否可以在不编写生成器函数的情况下从链表中进行迭代？

2 个答案: