python3----练习题(三级菜单)

三级菜单程序
. 运行程序输出第一级菜单
. 选择一级菜单某项，输出二级菜单，同理输出三级菜单
. 退出时返回上一级菜单

menu = {
    '北京市': {
        '东城区': {
            'aa',
            'bb',
        },
        '西城区': {
            'cc',
            'dd',
        },
    },
    '天津市': {
        '和平区': {
            'ee',
            'ff',
        },
        '河东区': {
            'gg',
            'hh',
        },
    },
    '河北省': {
        '石家庄市': {
            '长安区': {
                'jj',
                'kk',
            },
            '桥西区': {
                'll',
                'mm',
            }
        },
        '唐山市': {
            '路南区',
            '路北区',
        },
    }
}
# print(menu)
# menu = menu['北京']
# print(menu)
# menu = menu['东城区']
# print(menu)
level = []  # 保存上一级菜单,用于回退
while True:
    for key in menu:
        print(key)
    choice = input('请选择,退出请输入b:').strip()
    if choice == 'b':
        if len(level) == 0:
            break
        menu = level[-1]
        level.pop()
        continue
    if len(choice) == 0 or choice not in menu:
        continue
    level.append(menu)
    menu = menu[choice]