elif使用字典，但不超过值[重复]

Question

大家好，假设我想用字典替换一个大的el-ifs语句。简化版本如下：

def functionA(a1, a2, a3):

    results = {
        'case1': a2/a3,
        'case2': 1,
        ...
    }
    return results[a1]

所以a1将是一个字符串（'case1'或'case2'或...），问题是在某些情况下，a3可能是0，所以结果字典无法定义（在所有这些情况下，a1都不是'case1'）。例如：

functionA('case2', 1.0, 3.0)
Out[81]: 1
functionA('case2', 1.0, 0.0)
ZeroDivisionError: integer division or modulo by zero

所以在第二种情况下，我期望1，但我得到了一个错误。

一种可能的解决办法是：

def functionA(a1, a2, a3):
    results = {
        'case1': str(a2) + '/' + str(a3),
        'case2': str(1),
    }

    return eval(results[a1])

由于我有一个复杂计算的多个案例，有没有更好的解决方案？

Answer 1

如果只是一个值，你可以用一个简单的If-else。

def functionA(a1, a2, a3):

     if a3:
        case1_value = a2/a3
     else: 
        case1_value = a2

    results = {
        'case1': case1_value,
        'case2': 1,
        ...
    }
    return results[a1]

或者使用这个if-else条件创建一个dvision函数，并在dict值中使用它。

Answer 2

您可以在字典中使用三元运算符

def functionA(a1, a2, a3):

    results = {
        'case1': a2 if a3==0 else a2/a3,
        'case2': 1
    }
    return results[a1]

链接：https://repl.it/NRkd

Answer 3

def functionA(a1, a2, a3):
    results = {
        'case1': a2 / a3 if a3 != 0.0 else a2,
        'case2': 1,
        # ...
    }
    return results[a1]

然而，我建议不要采用这种方法，因为整个字典必须首先计算，只有一个值。

相反，如果你真的想使用字典，并且它也很有效，我建议：

myproc = {
    'case1': lambda a2, a3: a2 / a3 if a3 != 0.0 else a2,
    'case2': lambda a2, a3: 1
}

然后：

>>> myproc['case1'](2, 3)
0.6666666666666666