Question

对于看似基本的问题感到抱歉，但我无法在任何地方找到它。在Python 2中，我想将1变量函数应用于它自己的输出，存储所有步骤的列表，即如果f（x）返回x * x然后从2迭代，我需要得到

[2, 4, 16, 256, 65536, ...]

理想情况下，我需要传递我的函数f，第一个输入1，以及我希望保留的迭代次数。

我想这在某种意义上与reduce相反，与函数式编程中的unfold有些相似。

一种天真的方法是写

out = [2] for x in xrange(5): out.append(f(out[-1]))

有什么好的Pythonic方法呢？非常感谢你。

Answer 1

您需要的是＆＃34; Generator＆＃34;。例如，

def f(x, n):
    for _ in range(n):
        yield x
        x = x * x

l = list(f(2, 5))

print(l)             # [2, 4, 16, 256, 65536]

或者

def f(x):
    while True:
        yield x
        x = x * x

for v in f(2):
    if v > 100000:
        break
    print(v),        # 2 4 16 256 65536

Answer 2

理想情况下，我需要传入我的函数f，第一个输入1和我希望保留的迭代次数。

这是一个展开函数，它接受函数，起始值和迭代计数。

def unfold(function, start, iterations):
    results = []
    for _ in range(iterations):
        results.append(start)
        start = function(start)

    return results

您可以按预期使用：

>>> print unfold(lambda x: x*x, 2, 5)
[2, 4, 16, 256, 65536]

在Python中生成自己输出的迭代列表

2 个答案: