python——字典小应用

像下面的代码可以写成另一种模式

def home():
    print('欢迎进入Home界面')

def finance():
    print('123')

def book():
    print('456')
#===========================以下代码可以替换成最的下面的
#执行过程
print('1. home')
print('2. finance')
print('3. book')
while True:
    key_num = input('请输入您要进入页面前的数字：')
    if key_num == '1':
        home()
    elif key_num == '2':
        finance()
    elif key_num == '3':
        book()

下面的代码把函数名做为字典的值，数字做为键，如果要执行某个函数的时候直接在字典里面取就可以了，还可能判断一下，输入的键是不是在字典里。

option = """
    ========= OPTION ===========
    1. home
    2. finance
    3. book  
"""
option_dic = {
    '1': home,
    '2': finance,
    '3': book
}

exit_flag = False
while not exit_flag:
    print(option)
    key_num = input('请输入您要进入页面前的数字：')
    if key_num in option:
        option_dic[key_num]()
    else:
        print('输入有误')