学生管理系统（1）

    task:
        学生管理系统：
        1.添加学生
        2.展示学生
        3.查询学生
        4.删除学生
        5.修改学生
        6.退出系统
    version:1.0
    author:ZXQ
    date:20-05-15

# coding='utf-8'

# 定义查询姓名所在列表所在索引函数
# 以列表形式返回索引位置
def get_name_list(name, students):
    length = len(students)
    name_list = []
    for i in range(length):
        if name in students[i]:
            name_list.append(i)

    return name_list


# 学生管理系统函数
def stu_manager():
    students = []
    print('-' * 20)
    print("""学生管理系统：
                1.添加学生 
                2.展示学生
                3.查询学生 
                4.删除学生 
                5.修改学生 
                6.退出系统 """)
    print('-' * 20)

    while True:
        choice = input('请输入您要选择的操作（1-6）：')

        # 添加学生
        if choice == '1':
            name = input('请输入学生姓名：')
            age = input('请输入学生年龄：')
            ID = input('请输入学生学号：')
            stu = [name, age, ID]
            students.append(stu)
            print('添加成功！')

        # 展示学生
        elif choice == '2':
            length = len(students)
            for i in range(length):
                print(students[i])

        # 查询学生
        elif choice == '3':
            name = input('请输入要查询的学生姓名：')
            name_list = get_name_list(name, students)
            if name_list:
                for n in name_list:
                    print('您查询的学生信息为：\n \
                    姓名：%s\t 年龄：%s \t 学号：%s' % (students[n][0], students[n][1], students[n][2]))
            else:
                print('该学生不存在，请检查！')

        # 删除学生
        elif choice == '4':
            name = input('请输入要删除的学生姓名：')
            name_list = get_name_list(name, students)
            if len(name_list):
                for n in name_list:
                    students.remove(students[n])
                    print('删除成功！')
            else:
                print('该学生不存在！')

        # 修改学生
        elif choice == '5':
            old_name = input('请输入要修改的学生：')
            name_list = get_name_list(old_name, students)
            if name_list:
                for n in name_list:
                    new_name = input('请输入学生姓名：')
                    age = input('请输入学生年龄：')
                    ID = input('请输入学生学号：')
                    new_stu = [new_name, age, ID]
                    students[n] = new_stu
                    print('修改成功！')
            else:
                print('该学生不存在，请检查！')

        # 退出系统
        elif choice == '6':
            print('即将退出系统，欢迎下次使用！')
            exit()
        else:
            input('您的输入有误，请重新输入：')


if __name__ == '__main__':
    stu_manager()