三级菜单练习

实现功能

• 输入对应名称能够进入下一级菜单
• 可以返回上一层菜单

#菜单
menu = {
    '北京' : {
        '海淀' : {
            '五道口' : {
                'soho' : {
                    'ceo' : {},
                    'coo' : {}
                },
                '网易' : {},
                'google' : {}
            },
            '中关村' : {
                '爱奇艺' : {},
                '汽车之家' : {},
                'youku' : {}
            },
            '上地' : {
                '百度' : {}
            }
        },
        '昌平' : {
            '沙河' : {
                '老男孩' : {},
                '北航' : {}
            },
            '天通苑' : {},
            '回龙观' : {}
        },
        '朝阳' : {},
        '东城' : {}
    },
    '上海' : {
        '闵行' : {
            '人民广场' : {
                '炸鸡店' : {}
            }
        },
        '闸北' : {
            '火车站' : {
                '携程' : {}
            }
        },
        '浦东' : {}
    },
    '山东' : {}
}

代码实现

current_layer = menu
layers = []

while True:
    for k in current_layer:
        print(k)

    choice = input('>>:').strip()

    if not choice : continue

    if choice in current_layer:
        layers.append(current_layer)
        current_layer = current_layer[choice]
    elif choice == 'b':
        if len(layers) != 0 :
            current_layer = layers.pop()
        else:
            print('已经在顶层了')