求大佬帮忙看看如何修改
bus = ['龙江新城市', '阳光广场', '汉江路', '嫩江路', '清凉山公园', '拉萨路', '五台山', '莫愁路']
dic = {'龙江新城市': ['白云园', '腾飞园'], '阳光广场': ['龙江小区', '芳草园'], '汉江路': ['金信花园', '龙凤花园'], '嫩江路': ['西城蓝湾', '花开四季', '银花东区'], '清凉山公园': ['成业家园'], '拉萨路': ['书香门第'],'五台山': ['美晨雅阁', '枣园小区'], '莫愁路': ['桃园小区', '南天东区', '桂苑']}
while True:
start = input("请输入起点小区名:")
for k, v in dic.items():
if start in v:
start = k
break
else:
print('找不到你的小区,请重新输入!')
start = input("请输入起点小区名:")
end = input("请输入终点小区名:")
for k, v in dic.items():
if end in v:
end = k
break
start_index = bus.index(start)
end_index = bus.index(end)
stops = end_index - start_index
if stops > 0:
print('起点站:{}站,终点站:{}站,共{}站。'.format(start, end, abs(stops)))
else:
print('您需要乘坐反方向线路,从{}站前往{}站需要{}站路。'.format(start, end, abs(stops)))
repeat = input('是否退出查询?')
if repeat.upper() == 'E':
print('查询结束,退出程序!')
break