以功能样式制作的联结功能
问题内容:
最近,在阅读Python “ Functional Programming
HOWTO”时
,我遇到了一个提到的test_generators.py
标准模块,在其中找到了以下生成器:
# conjoin is a simple backtracking generator, named in honor of Icon's
# "conjunction" control structure. Pass a list of no-argument functions
# that return iterable objects. Easiest to explain by example: assume the
# function list [x, y, z] is passed. Then conjoin acts like:
#
# def g():
# values = [None] * 3
# for values[0] in x():
# for values[1] in y():
# for values[2] in z():
# yield values
#
# So some 3-lists of values *may* be generated, each time we successfully
# get into the innermost loop. If an iterator fails (is exhausted) before
# then, it "backtracks" to get the next value from the nearest enclosing
# iterator (the one "to the left"), and starts all over again at the next
# slot (pumps a fresh iterator). Of course this is most useful when the
# iterators have side-effects, so that which values *can* be generated at
# each slot depend on the values iterated at previous slots.
def simple_conjoin(gs):
values = [None] * len(gs)
def gen(i):
if i >= len(gs):
yield values
else:
for values[i] in gs[i]():
for x in gen(i+1):
yield x
for x in gen(0):
yield x
我花了一段时间才了解它是如何工作的。它使用一个可变列表values
来存储迭代器的结果,并且N + 1迭代器返回values
,该值遍历整个迭代器链。
当我在阅读有关函数式编程的过程中偶然发现此代码时,我开始考虑是否可以使用函数式编程(使用itertools
模块中的函数)重写此联合生成器。有很多(在刚刚结束一目了然写函数式程序的这个文章中的食谱节)。
但是,不幸的是,我还没有找到任何解决方案。
那么,是否有可能仅使用itertools
模块就可以使用功能编程来编写此联合生成器?
谢谢
问题答案:
这似乎可行,但仍然很懒:
def conjoin(gs):
return [()] if not gs else (
(val,) + suffix for val in gs[0]() for suffix in conjoin(gs[1:])
)
def range3():
return range(3)
print list(conjoin([range3, range3]))
输出:
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
显示可变状态的示例用法:
x = ""
def mutablerange():
global x
x += "x"
return [x + str(i) for i in range(3)]
print list(conjoin([range3, mutablerange]))
输出:(观看不断增加的“ x”数)
[(0, 'x0'), (0, 'x1'), (0, 'x2'), (1, 'xx0'), (1, 'xx1'), (1, 'xx2'), (2, 'xxx0'), (2, 'xxx1'), (2, 'xxx2')]
如果我们使用itertools.product
:
x = ""
print list(itertools.product(range3(), mutablerange()))
结果如下:
[(0, 'x0'), (0, 'x1'), (0, 'x2'), (1, 'x0'), (1, 'x1'), (1, 'x2'), (2, 'x0'), (2, 'x1'), (2, 'x2')]
因此,可以清楚地看到,它itertools.product
缓存了迭代器返回的值。