python - 如何编写迭代器？

Question

我有一个节点网络，在节点之间传递结构化数据。对于我的子问题，我们有这个分支——节点的线性序列:

nodes = [source, n1, n2, n3, n4]

第一个节点是一个生成器，每个其他节点从输入节点获取值并提供输出值。当前的实现是普通的 get() 从管道和 put() 到管道，每个节点都有单独的线程(这是有原因的)。我想将其更改为 yield 迭代器。

我想通过以下方式进行评估(如果我们认为节点是可调用的):

for result in n4(n3(n2(n1(source()))):
    print result

我想象这样构建评估上下文:

context = src
for node in nodes[1:]:
    context = pipe(context, node)

for result in context:
    print result

限制:

我仍然希望能够单独使用节点 - 而不是嵌套，通过其他方式管道传输数据，因为节点可能位于单独的线程中。示例:[source, n1,n2] 在一个线程中(可能嵌套)，[n3, n4] 在另一个线程中(可能嵌套)，数据在 n2 和 n3。案例:可能有一个非线性节点图，我想以这种方式对分支进行分组。

node 必须是一个类来保存计算状态

context 和 pipe(context, node) 的实现可能是什么样子的？或者如果可以用不同的方式解决，您有什么提示吗？

可以在 Python 3.3 中yield from (PEP380)以任何方式帮助我的案子？

Answer 1

如果您只想组合任意数量的函数(或可调用对象)，请使用 compose_mult recipe来自 functional模块文档。

使用它的解决方案:

from functional import compose, foldr, partial
from itertools  import imap
compose_mult = partial(reduce, compose) 
chain_nodes = lambda nodes: imap(compose_mult(nodes[1:]), nodes[0])
chain_gen_nodes = lambda nodes: imap(compose_mult((g.send for g in nodes[1:])), nodes[0])


# equivalent not as a one-liner
#def chain_nodes(nodes):
#    source = nodes[0]
#    composed_nodes = compose_mult(nodes[1:])
#    return (composed_nodes(x) for x in source)

如果节点是接受输入的生成器(通过 send )，则使用 chain_gen_nodes ，提取它们的发送函数。

但是请注意，不允许send到刚刚启动的生成器(因为它必须在 yield 点才能接收值)。这是您必须自己处理的事情，例如让您的发电机 yield在他们的第一次迭代中使用虚拟值，并在将它们发送到 chain_nodes 之前在某个时刻推进它们.或者您可以将您的节点保留为普通的可调用对象。

如果您确实需要将迭代器推进一步:next(izip(*nodes[1:]))