Python：为此做的更有效的方法

Question

我是 python 的初学者，我想知道是否有一种优雅的方法来做到这一点：

if societe_var == 'apple':
    bu_list = bu_list_apple
elif societe_var == 'banana':
    bu_list = bu_list_banana
elif societe_var == 'pear' :
    bu_list = bu_list_pear
else :
    bu_list = bu_list_cherry

此致，

凯尔0

Answer 1

使用字典：

bu_list_map = {
    'apple': bu_list_apple,
    'banana': bu_list_banana,
    'pear': bu_list_pear
}

bu_list = bu_list_map.get(societe_var, default=bu_list_cherry)

使用.get()方法的default参数来处理回退情况

Answer 2

将您可能的值放入 dict 并使用.get()获取其中之一（如果未找到则为默认值）：

bu_lists = {
    'apple': bu_list_apple,
    'banana': bu_list_banana,
    'pear': bu_list_pear,
}
bu_list = bu_lists.get(societe_var, bu_list_cherry)