Python:编写 n 个函数组合的更好方法？

Question

我写了一个函数“rep”，它接受一个函数 f 并接受 f 的 n 个组合。所以 rep(square,3) 的行为是这样的:square(square(square(x)))。当我将 3 传递给它时，rep(square,3)(3)=6561。

我的代码没有问题，但我想知道是否有办法让它“更漂亮”(或更短)而无需调用另一个函数或导入任何东西。谢谢!

def compose1(f, g):
    """Return a function h, such that h(x) = f(g(x))."""
    def h(x):
        return f(g(x))
    return h

def rep(f,n):
    newfunc = f
    count=1
    while count < n:
        newfunc = compose1(f,newfunc)
        count+=1
    return newfunc

Answer 1

如果您追求速度，for 循环显然是最佳选择。但是，如果您正在寻找理论上的学术认可 ;-)，请坚持使用简洁的功能习语。喜欢:

def rep(f, n):
    return f if n == 1 else lambda x: f(rep(f, n-1)(x))