compose functions in python

compose_functions.py

def compose(*funcs):
    """Functional composition

    [f, g, h] will be f(g(h(x)))
    """
    def compose_two(f, g):
        def c(x):
            return f(g(x))
        return c
    return functools.reduce(compose_two, funcs)

Example

In [3]: def f(n: int) -> int: return n + 1                                                                                                                                                                          

In [4]: def g(n: int) -> int: return n * 2                                                                                                                                                                          

In [5]: fg = compose(print, f, g)                                                                                                                                                                                   

In [6]: fg(2)                                                                                                                                                                                                       
5

In [7]: compose(f, g)(2)                                                                                                                                                                                            
Out[7]: 5