如何将嵌套的if/else函数调用转换为python中的字典？

Question

我有一个函数，它根据满足的条件语句调用不同的函数。该函数目前看起来如下所示。

def func1(var_a, var_b=1):
    return 'Some values'

def func2(var_c, var_d=2):
    return  'Some values'

def func3(var_e, var_f):
    return 'Some values'


def choose_function(a, b, c):
    if a == 'first_option':
        val = func1(b, c)
    elif a == 'second_option':
        val = func2(b, c)
    elif a == 'third_option':
        val = func3(b, c)

    return val

函数choose_function在将来可以有8-10个if else条件。有没有更好的方式来处理这类条件语句。

我是否可以将其转换为字典，该字典根据字典中作为键提供的变量a调用函数。更具体地说，我希望该函数如下所示

def choose_function(a, b, c):
    func_options ={
        'first_option': func1(b, c),
        'second_option': func2(b, c),
        'third_option': func3(b, c)
    }

    return func_options[a]

即使上面的函数工作正常，当有人提供字典键中不存在的值时，我如何处理这种情况？

Answer 1

存储函数引用-定义字典时不要调用函数

def choose_function(a, b, c):
    func_options ={
        'first_option': func1,
        'second_option': func2,
        'third_option': func3
    }

    return func_options[a](b,c)

使用字典上的.get()方法处理无效选项

    return func_options.get(a, lambda *args, **kwargs: ...)(b,c)

用打印/缺省值等替换...。