Python初级---学生系统

使用组合数据类型(列表，元组，字典，集合)中任意一种存储方式，完成一个简单的学生管理系统，可以多次录入学生信息到系统并完成基本的增加删除修改，查询操作。
学生信息： 学号，姓名，性别，年龄，专业；
注意：
单星号 * 用于对列表LIST或元组tuple中的元素进行取出(unpacke)
双星号 ** 可将字典里的“值”取出

def add_student(students, student_info):
    # 增加学生信息到系统
    students[student_info["学号"]] = student_info

def delete_student(students, student_id):
    # 删除学生信息
    if student_id in students:
        del students[student_id]
        print("删除成功！")
    else:
        print("学号不存在，删除失败！")

def modify_student(students, student_id, new_info):
    # 修改学生信息
    if student_id in students:
        students[student_id].update(new_info)
        print("修改成功！")
    else:
        print("学号不存在，修改失败！")

def query_student(students, student_id):
    # 查询学生信息
    if student_id in students:
        student_info = students[student_id]
        print("学号: ", student_info["学号"])
        print("姓名: ", student_info["姓名"])
        print("性别: ", student_info["性别"])
        print("年龄: ", student_info["年龄"])
        print("专业: ", student_info["专业"])
    else:
        print("学号不存在，查询失败！")

def main():
    students = {}  # 用字典存储学生信息，学号作为键

    while True:
        print("\n========= 学生管理系统 =========")
        print("1. 录入学生信息")
        print("2. 删除学生信息")
        print("3. 修改学生信息")
        print("4. 查询学生信息")
        print("5. 退出系统")
        choice = input("请输入对应的操作编号(1/2/3/4/5): ")

        if choice == "1":
            # 录入学生信息
            student_info = {}
            student_info["学号"] = input("请输入学号: ")
            student_info["姓名"] = input("请输入姓名: ")
            student_info["性别"] = input("请输入性别: ")
            student_info["年龄"] = input("请输入年龄: ")
            student_info["专业"] = input("请输入专业: ")
            add_student(students, student_info)
            print("学生信息录入成功！")

        elif choice == "2":
            # 删除学生信息
            student_id = input("请输入要删除的学号: ")
            delete_student(students, student_id)

        elif choice == "3":
            # 修改学生信息
            student_id = input("请输入要修改的学号: ")
            new_info = {}
            new_info["姓名"] = input("请输入新的姓名: ")
            new_info["性别"] = input("请输入新的性别: ")
            new_info["年龄"] = input("请输入新的年龄: ")
            new_info["专业"] = input("请输入新的专业: ")
            modify_student(students, student_id, new_info)

        elif choice == "4":
            # 查询学生信息
            student_id = input("请输入要查询的学号: ")
            query_student(students, student_id)

        elif choice == "5":
            # 退出系统
            print("退出学生管理系统，再见！")
            break

        else:
            print("输入的操作编号无效，请重新输入！")

if __name__ == "__main__":
    main()