Python 武侠游戏：选择门派代码优化

请问你想加入的门派编号是从1到4的吗？如果是，下面是修改后的代码：\n\nprint("欢迎来到这个武侠世界")\n门派 = int(input("请输入你想加入的门派的编号\n1.六扇门\n2.明教\n3.丐帮\n4.武当"))\nprint("你选择的门派是", 门派)\n\nif 门派 == 1:\n print("前往北平报道")\nelif 门派 == 2:\n print("前往西域光明顶报道")\nelif 门派 == 3:\n print("前往南直隶报道")\nelif 门派 == 4:\n print("前往湖广报道")\nelse:\n print("请输入有效的门派编号")