lambda_closure.py

def function_toute_prete(y):
    return y

def operation(n):
    """
    operation(n) is a function that adds n to its argument if n = 0 [2] or multiply n by its two arguments if n = 1 [2].
    
    For instance, add(3) : x -> x * 3
    """
    functions = [
        lambda x: n + x,
        lambda x,y : n * x * y,
        lambda x: n * x,
        lambda x: n / x,
        lambda y: function_toute_prete(y)
    ]
    
    if n < len(functions) - 1:
        return functions[n]
    else:
        return None

mul3 = operation(3)
print(mul3(3,2)) # Prints 18

add6 = operation(6)
print(add6(4)) # Prints 10