python - 如何在Python中编写类似Mma的NestList的函数

例如：
嵌套列表（f，x，3）--->[x，f（x），f（f（x）），f（f（x）]
http://reference.wolfram.com/mathematica/ref/NestList.html

最佳答案

你可以把它写成发电机：

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)