Python中的条件函数链接

Question

想象一下，我想通过链接子函数来实现g函数。这可以通过以下方式轻松完成：

def f1(a):
    return a+1

def f2(a):
    return a*2

def f3(a):
    return a**3

g = lambda x: f1(f2(f3(x)))

但是，现在考虑一下，哪些子功能将链接在一起，取决于条件：具体而言，是事先已知的用户指定选项。当然可以这样做：

def g(a, cond1, cond2, cond3):

    res = a
    if cond1:
        res = f3(res)
    if cond2:
        res = f2(res)
    if cond3:
        res = f1(res)
    return res

但是，每次调用函数时，我都不是动态地检查这些静态条件，而是假设最好根据其组成函数预先定义函数g。不幸的是，以下给出了RuntimeError: maximum recursion depth exceeded：

g = lambda x: x
if cond1:
    g = lambda x: f3(g(x))
if cond2:
    g = lambda x: f2(g(x))
if cond3:
    g = lambda x: f1(g(x))

有没有一种在Python中进行这种条件链接的好方法？请注意，要链接的函数可以是N，因此不能单独定义所有2 ^ N个函数组合（本例中为8）。

Answer 1

我找到了一个使用装饰器的解决方案。看一看：

def f1(x):
    return x + 1

def f2(x):
    return x + 2

def f3(x):
    return x ** 2


conditions = [True, False, True]
functions = [f1, f2, f3]


def apply_one(func, function):
    def wrapped(x):
        return func(function(x))
    return wrapped


def apply_conditions_and_functions(conditions, functions):
    def initial(x):
        return x

    function = initial

    for cond, func in zip(conditions, reversed(functions)):
        if cond:
            function = apply_one(func, function)
    return function


g = apply_conditions_and_functions(conditions, functions)

print(g(10)) # 101, because f1(f3(10)) = (10 ** 2) + 1 = 101

在定义g函数时，只检查一次条件，调用它们时不会检查它们。

Answer 2

我能想到的结构最相似的代码必须按照以下方式构建，你的f1.. f3将需要成为伪装饰器，如下所示：

def f1(a):
    def wrapper(*args):
        return a(*args)+1
    return wrapper

def f2(a):
    def wrapper(*args):
        return a(*args)*2
    return wrapper

def f3(a):
    def wrapper(*args):
        return a(*args)**3
    return wrapper

然后您可以将这些应用于每个功能。

g = lambda x: x
if cond1:
    g = f3(g)
if cond2:
    g = f2(g)
if cond3:
    g = f1(g)
g(2)

返回：

# Assume cond1..3 are all True
17 # (2**3*2+1)