Повторите композицию функции n раз в Python, как в Haskell. - PullRequest
Новая
       43

Повторите композицию функции n раз в Python, как в Haskell.

1 голос
/ 28 февраля 2020

Этот код НЕ работает: 

def inc(x):
    return x + 1

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

inc_10 = repeat(inc, 10)

#TypeError: unsupported operand type(s) for +: 'function' and 'int'


'''
# Ideally
print(inc_10(0))
# 10
'''

Как я могу написать его более Pythoni c способом или способом лямбда-исчисления?

1 Ответ

3 голосов
/ 28 февраля 2020

Вам все равно нужно возвращать функцию, а не результат вызова f, в рекурсивном случае. 

# repeat :: (a -> a) -> Integer -> a -> a
# repeat _ 0 = id
# repeat f n = \x -> f (repeat f (n-1) x)
def repeat(f, n):
    if n == 0:
        return lambda x: x
    else:
        return <b>lambda x:</b> f (repeat(f, n-1)<b>(x)</b>)

Немного легче читать, если вы также определите составную функцию: 

def identity(x):
    return x


def compose(f, g):
    return lambda x: f(g(x))


# repeat :: (a -> a) -> Integer -> (a -> a)
# repeat _ 0 = id
# repeat f n = f . repeat f (n - 1)
def repeat(f, n):
    if n == 0:
        return identity
    else:
        return compose(f, repeat(f, (n-1)))

Или используя functools.reduce: 

# repeat :: (a -> a) -> Integer -> (a -> a)
# repeat f n = foldr (.) id $ replicate n f
def repeat(f, n):
    return reduce(compose, [f]*n, identity)
...