python - 迭代函数的 Python 迭代器

Question

所以我必须制作以下函数->迭代。在第一次调用时，它应该在第二个 func 和第三个 func.func 上返回标识。知道怎么做吗？我尝试查看iter和下一个方法 buf 失败：（

>>> def double(x):
        return 2 * x

>>> i = iterate(double)
>>> f = next(i)
>>> f(3)
3
>>> f = next(i)
>>> f(3)
6
>>> f = next(i)
>>> f(3)
12
>>> f = next(i)
>>> f(3)
24

Answer 1

可能是这样的：

>>> import functools
>>> def iterate(fn):
    def repeater(arg, _count=1):
        for i in range(_count):
            arg = fn(arg)
        return arg
    count = 0
    while True:
        yield functools.partial(repeater, _count=count)
        count += 1


>>> i = iterate(double)
>>> f, f2, f3, f4 = next(i), next(i), next(i), next(i)
>>> f(3), f2(3), f3(3), f4(3)
(3, 6, 12, 24)
>>> f(3), f2(3), f3(3), f4(3)
(3, 6, 12, 24)

因此，您编写了一个函数，该函数以指定为参数的次数调用原始函数，并预先绑定了 count 参数。