Question

我嵌套了if-else语句，每个条件调用一个不同的函数。像 -

var = "somestring"
if var == 'call':
    if var2 == 'someothercomparison':
        return func1(arg1, arg2)
    elif var2 == 'otherway':
        return func2()
elif var == 'sms':
    return func3(arg3)
elif var == 'mail':
    return func4(arg4)

这只是一个示例数据..还有更多elif。

如果我尝试使用字典，那么它将变得低效，因为编译器/解释器将尝试执行所有函数来验证字典语法。所以我不能使用字典。

还有其他优雅/优雅/ pythonic方式来完成这项任务吗？

Answer 1

你可以使用带有函数和参数的dicts的dict作为值：

def func1(s):print(s)
def func2(s):print(s)
def func3(s):print(s)
def func4(s):print(s)

d = {"call":{'someothercomparison': (func1,"arg1"),'otherway':(func2,"arg2")},"sms":(func3,"arg3"),"mail":(func4,"func4")}

var = "call"
var2 = 'otherway'

def main(var,var2):
    f = d.get(var, {}).get(var2)
    if f:
       return  f[0](f[1])
    f = d.get(var)
    if f:
        return  f[0](f[1])
    return whatever

Answer 2

您可以使用词典和lambdas的组合：

options={"a":lambda:doA(arg1,arg2),"b":lambda:doB(arg3,arg4)}
options[var]()

如果您需要默认值：

from collections import defaultdict
options=defaultdict(lambda:lambda:doDefault(arg1,arg2),options)

做分支，只需让lambda这样做：

def branch1():
    if cond:
      doA()
    else:
      doElse(args)
options={"abc":lambda:branch1()}

除了字典之外，Python中的Switch-Case替代方案

2 个答案: