7

是否有一个内置函数或标准库函数大致相当于

def recur_until(start, step_fu, stop_predicate=lambda _: False):
    current = start
    while not stop_predicate(current):
        yield current
        current = step_fu(current)

或者

def recur_while(start, step_fu, predicate=lambda _: True):
    current = start
    while predicate(current):
        yield current
        current = step_fu(current)

甚至只是

def recur(start, step_fu):
    current = start
    while True:
        yield current
        current = step_fu(current)

在任何版本的 Python 中？（后者与 结合使用时与其他两个一样好itertools.takewhile。）

像这样的生成器函数将允许迭代地计算某些递归定义的序列，即一阶递归关系。

虽然这些在需要时实施起来并不难，但我觉得像它们这样的东西应该是itertools或可能functools的一部分，但如果是的话，我还无法在文档中发现它。

使用示例：

list(recur_until(2, lambda x: x**2 - 1, lambda x: x > 1e4))
# [2, 3, 8, 63, 3968]

也应该与非数字元素一起使用：

list(recur_until('', lambda x: '[{}]({})'.format(x, len(x)), lambda x: len(x) > 30))
# ['',
#  '[](0)',
#  '[[](0)](5)',
#  '[[[](0)](5)](10)',
#  '[[[[](0)](5)](10)](16)',
#  '[[[[[](0)](5)](10)](16)](22)']
4

3 回答 3

5

在 Python 3.3+ 中，newitertools.accumulate可以与其他 itertools 结合用于此目的

例如：

>>> from itertools import accumulate, repeat, takewhile
>>> fun = accumulate(range(2, 10), lambda x, _: x**2 - 1)
>>> list(fun)
[2, 3, 8, 63, 3968, 15745023, 247905749270528, 61457260521381894004129398783]
>>> fun = takewhile(lambda y: y < 1e4, accumulate(repeat(2), lambda x, _: x**2 - 1))
>>> list(fun)
[2, 3, 8, 63, 3968]

accumulate接受一个序列和一个带有 2 个参数的函数：第一个是累加值，第二个是序列中的下一个值。在这种情况下，我们只需要第一个参数，它将是第accumulate一次调用传递函数时传递给的序列的第一个元素，以及后续调用时该函数的返回值。

因此，我们只需要传递序列的开头作为我们的初始值——2在这种情况下。序列其余部分的内容无关紧要，但我们可以使用它的长度来控制我们想要多少元素（如第一个示例）或创建一个无限生成器（如第二个示例）。

于 2016-02-26T17:44:31.743 回答
4

缺少的链接是您需要一些东西来将递归步进函数转换为生成器。一旦你有了它，你就可以使用任何 itertools 方法。

def recur_to_gen(step_fu, current, sentinel=None):
    while current != sentinel:
        yield current
        current = step_fu(current)


matches = itertools.takewhile(predicate, recur_to_gen(step_fu, start))

recur_to_gen建议添加到 . 可能是合理的事情itertools。

于 2016-02-26T16:54:57.117 回答
3

功能包提供了模拟这一点的部分。

from functional import dropWhile, iterate    
recur = dropWhile(predicate, iterate(step_func, start))

例如，

>>> next(dropWhile(lambda x : x < 10, iterate(lambda x: x + 1, 2)))
10

（dropWhile实际上和 没有什么不同itertools.dropwhile。）

于 2016-02-26T17:03:28.710 回答