python - 重复函数python

Question

我被 python 中的高阶函数困住了。我需要编写一个重复函数 repeat，它在给定参数 x 上应用函数 f n 次。

例如，repeat(f, 3, x) 是 f(f(f(x)))。

这是我的:

def repeat(f,n,x):
    if n==0:
        return f(x)
    else:
        return repeat(f,n-1,x)

当我尝试断言以下行时:

plus = lambda x,y: repeat(lambda z:z+1,x,y)
assert plus(2,2) == 4

它给了我一个AssertionError。我读到了 How to repeat a function n times但我需要以这种方式完成，但我无法弄清楚...

Answer 1

你有两个问题:

1. 你递归的次数错误(如果 n == 1，函数应该被调用一次)；和
2. 您没有对递归调用的返回值调用 f，因此该函数只会应用一次。

尝试:

def repeat(f, n, x):
    if n == 1: # note 1, not 0
        return f(x)
    else:
        return f(repeat(f, n-1, x)) # call f with returned value

或者，或者:

def repeat(f, n, x):
    if n == 0:
        return x # note x, not f(x)
    else:
        return f(repeat(f, n-1, x)) # call f with returned value

(感谢 @Kevin 的后者，它支持 n == 0)。

例子:

>>> repeat(lambda z: z + 1, 2, 2)
4
>>> assert repeat(lambda z: z * 2, 4, 3) == 3 * 2 * 2 * 2 * 2
>>>