如何实现函数的变体/变体

Question

这是我的第一个问题，因此对于我确信我会犯的任何发布错误，我深表歉意。我确实首先搜索了以前回答的问题，但我没有找到合适的解决方案，尽管我知道我可能没有使用正确的术语或关键字进行搜索。

我想在 Python 中快速评估一个通用函数。它被传递给优化器（fmin_cobyla），我希望它只接受一个参数。我希望能够选择使用可能由标志控制的此功能的不同变体（变体？口味？）。由于我希望快速评估它，因此在函数中包含一堆 if 或 case 语句是没有意义的（对吗？），所以目前我有围绕函数变体定义的 if 语句：

if flag==1:
    def f(x):
        for i in range(0,len(x)*3,3):
            A[i:i+3,i:i+3]=1/x[i/3]*np.eye(3)
        tmp=np.linalg.solve(A,b)
        y=somecmodulefunction(tmp)
        return y
elif flag==2:
    def f(x):
        for i in range(0,len(x)*3,3):
            A[i:i+3,i:i+3]=x[i/3]*np.eye(3)
        tmp=np.linalg.solve(A,b)
        y=somecmodulefunction(tmp)
        return y
...

除了一行之外，功能相同。这是正确的方法吗？有没有办法做到这一点，如果我有 10 个不同的变体，它不仅仅是一个长而笨重的代码块？

谢谢！

Answer 1

为什么不将标志变量也作为输入传递给函数？

def f(x, flag):
    for i in range(0,len(x)*3,3):
        if flag == 1:
           A[i:i+3,i:i+3]=1/x[i/3]*np.eye(3)
        elif flag == 2:
           A[i:i+3,i:i+3]=x[i/3]*np.eye(3)
    tmp=np.linalg.solve(A,b)
    y=somecmodulefunction(tmp)
    return y

如果有一个默认值为flag，你可以使用，Default Argument Values：

def f(x, flag=1):
    for i in range(0,len(x)*3,3):
        if flag == 1:
           A[i:i+3,i:i+3]=1/x[i/3]*np.eye(3)
        elif flag == 2:
           A[i:i+3,i:i+3]=x[i/3]*np.eye(3)
    tmp=np.linalg.solve(A,b)
    y=somecmodulefunction(tmp)
    return y

因此，当您这样做时，f(10) 函数 f() 将使用等于 1 的 flag 执行。

如果您想使用另一个 flag 值，您可以简单地执行：f(10,2)。

Answer 2

尽量少重复自己。这意味着仅指定差异，即该表达式的反转。其余的都是一样的。如果您想更改某项计算的方式，则只需在一处进行更改。

def f(x, flag):
    for i in range(0,len(x)*3,3):
        temp = x[i/3]*np.eye(3)
        if flag-2:
            temp = 1/temp
        A[i:i+3,i:i+3] = temp
    tmp=np.linalg.solve(A,b)
    y=somecmodulefunction(tmp)
    return y

当flag 为2 时，if 分支将不会执行，因为2-2 等于0，这是一个虚假值。当flag 为1 时，将执行if 分支，因为1-2 等于-1，一个真实值。但是，我建议使用布尔标志（即，期望值为 True 或 False），因此您可以只使用 if flag: 并使用 True 表示您想要反转。