python - 如何在 Python 中编写像 Mma 的 NestList 这样的函数

Question

例如： NestList(f,x,3) ----> [x, f(x), f(f(x)), f(f(f(x)))]

http://reference.wolfram.com/mathematica/ref/NestList.html

Answer 1

你可以把它写成一个生成器：

def nestList(f,x,c):
    for i in range(c):
        yield x
        x = f(x)
    yield x

import math
print list(nestList(math.cos, 1.0, 10))

或者，如果您希望将结果作为列表，您可以在循环中追加：

def nestList(f,x,c):
    result = [x]
    for i in range(c):
        x = f(x)
        result.append(x)
    return result

import math
print nestList(math.cos, 1.0, 10)

Answer 2

def nest_list(f, x, i):
        if i == 0:
            return [x]
        return [x] + nest_list(f, f(x), i-1)


def nest_list(f, x, n):
    return [reduce(lambda x,y: f(x), range(i), x) for i in range(n+1)]

I found another way did it!

Answer 3

使用functional模块。它有一个名为 的函数scanl，它产生一个归约的每个阶段。然后，您可以减少f.