如何在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)