-- coding: utf-8 --

""" Created on Mon Dec 5 08:36:06 2022

@author: shipeilong """

import pandas as pd from datetime import datetime

pandas显示设置

#显示所有列

pd.set_option('display.max_columns', None)

#显示所有行

pd.set_option('display.max_rows', None)

#设置value的显示长度为100，默认为50

pd.set_option('max_colwidth',100)

格式输出

def print_message(*message): print("【{}】 【{}】".format(datetime.now(), message))

读取stu.csv文件

def read_stu_csv(): stus = pd.read_csv("stu.csv", index_col= False)

return stus 

查询全部学生信息

def read_all_stus(): stus = read_stu_csv() display(stus)

read_all_stus()

通过姓名查询某个学生信息

def read_someone_by_name(): stus = read_stu_csv() display(stus) stu_names = stus['姓名'].values.tolist()

while True:
    print_message("请输入查询学生姓名：")
    stu_name = input("姓名：")
    if stu_name not in stu_names:
        print_message("学生不存在，请重新输入！")
        continue
    else:
        stu = stus.loc[(stus['姓名'] == stu_name),:]
        print_message("【{}】学生信息查询成功！".format(stu_name))
        display(stu)
        break

read_someone_by_name()

通过电话查询某个学生信息

def read_someone_by_phone(): stus = read_stu_csv() display(stus) stu_phones = stus['电话'].values.tolist()

while True:
    print_message("请输入查询学生电话：")
    stu_phone = input("电话：")
    if int(stu_phone) not in stu_phones:
        print_message("学生不存在，请重新输入！")
        continue
    else:
        stu = stus.loc[(stus['电话'] == int(stu_phone)),:]
        print_message("【{}】学生信息查询成功！".format(stu_phone))
        display(stu)
        break

read_someone_by_phone()

通过姓名/电话查询某个学生信息

def read_stu(): stus = read_stu_csv() stu_names = stus['姓名'].values.tolist() stu_phones = stus['电话'].values.tolist() while True: stu_input = input("请输入学生姓名/电话：") if stu_input in stu_names: stu = stus.loc[(stus['姓名'] == stu_input),:] display(stu) break if int(stu_input) in stu_phones: stu = stus.loc[(stus['电话'] == int(stu_input)),:] display(stu) break else: print_message("未查询到【{}】信息，请重新输入".format(stu_input)) continue read_stu()

增加某个学生信息

def add_stu(): stus = read_stu_csv() stu_names = stus['姓名'].values.tolist() stu_phones = stus['电话'].values.tolist() while True: # 添加学生姓名 stu_name_flag = True while stu_name_flag: stu_name = input("请输入要添加的学生姓名：") if stu_name == "": print_message("学生姓名输入不能为空，请重新输入！") continue else: stu_name_flag = False # 添加学生电话 stu_phone_flag = True while stu_phone_flag: stu_phone = input("请输入该学生电话：") if stu_phone == "": print_message("学生电话输入不能为空，请重新输入！") continue else: stu_phone_flag = False # 添加学生生日 stu_birth_flag = True while stu_birth_flag: stu_birth = input("请输入该学生生日（格式：2000/1/1）：") if stu_birth == "": print_message("学生生日输入不能为空，请重新输入！") continue else: stu_birth_flag = False # 添加学生入学年份 year_flag = True while year_flag: year = input("请输入该学生入学年份（默认为2020）：") if year == "": print_message("输入为空，默认学生入学年份为2020！") year = 2020 year_flag = False else: year_flag = False

    last_id = stus.iloc[-1].to_dict()["学号ID"]
    stu_dict = {"学号ID":str(int(last_id)+1),"姓名":stu_name,"电话":stu_phone,"生日":stu_birth,"入学年份":year}
    stu_df = pd.DataFrame(stu_dict,index=[0])
    new_stus = stus.append(stu_df, sort=False)
    new_stus.to_csv("stu.csv", index = False)
    print_message("添加学生【{}】成功".format(stu_name))
    display(stu_df)
    break

add_stu()

#通过姓名修改某个学生姓名 def update_by_name(): stus = read_stu_csv() display(stus) stu_names = stus['姓名'].values.tolist() while True: stu_name = input("请输入待添加学生姓名") if stu_name not in stu_names: print_message("学生不存在，请重新输入！") continue else: stu = stus.loc[(stus['姓名'] == stu_name),:] display(stu)

        name_flag = True
        while name_flag:
            name = input("请输入修改姓名：")
            if name == "":
                print_message("修改姓名不能为空，请重新输入！")
                continue
            else:
                name_flag = False
    
    add_flag = True
    while add_flag:
        confirm = input("【{}】修改为【{}】元，请确认：（y/n）".format(stu_name, name))
        if confirm == "y":
            # 获取”姓名“index
            stu_index = list(stus.columns).index('姓名')
            # 获取学生index
            name_index = stus.loc[(stus['姓名'] == stu_name),:].index[0]
            # 获取学生原姓名
            stu_name_dict = stu.to_dict()["姓名"]
            stu_name_index = list(stu_name_dict.keys())[0]
            stu_name = stu_name_dict[stu_name_index]
            # 增加学生添加姓名
            stus.iloc[name_index,stu_index] = name
            # 写入stu表中(stu.csv)
            stus.to_csv("stu.csv", index = False)
            print_message("【{}】添加【{}】成功".format(stu_name, name))
            stu = stus.loc[(stus['姓名'] == name),:]
            display(stu)
            print_message("最新学生联系管理系统学生名单如下：")
            display(stus)
            break
        elif confirm == "n":
            print_message("取消添加！")
            break
        else:
            print_message("输入错误，请重新输入！")
            continue
    break

通过电话修改某个学生电话

def update_by_phone(): stus = read_stu_csv() display(stus) stu_phones = stus['电话'].values.tolist() while True: stu_phone = input("请输入待添加学生电话") if int(stu_phone) not in stu_phones: print_message("学生不存在，请重新输入！") continue else: stu = stus.loc[(stus['电话'] == int(stu_phone)),:] display(stu)

        phone_flag = True
        while phone_flag:
            phone = input("请输入修改电话：")
            if phone == "":
                print_message("修改电话不能为空，请重新输入！")
                continue
            else:
                phone_flag = False
    
    add_flag = True
    while add_flag:
        confirm = input("【{}】修改为【{}】元，请确认：（y/n）".format(stu_phone, phone))
        if confirm == "y":
            # 获取”电话“index
            stu_index = list(stus.columns).index('电话')
            # 获取学生index
            phone_index = stus.loc[(stus['电话'] == int(stu_phone)),:].index[0]
            # 获取学生原电话
            stu_phone_dict = stu.to_dict()["电话"]
            stu_phone_index = list(stu_phone_dict.keys())[0]
            stu_phone = stu_phone_dict[stu_phone_index]
            # 增加学生添加电话
            stus.iloc[phone_index,stu_index] = phone
            # 写入stu表中(stu.csv)
            stus.to_csv("stu.csv", index = False)
            print_message("【{}】添加【{}】成功".format(stu_phone, phone))
            stu = stus.loc[(stus['电话'] == phone),:]
            display(stu)
            print_message("最新学生联系管理系统学生名单如下：")
            display(stus)
            break
        elif confirm == "n":
            print_message("取消添加！")
            break
        else:
            print_message("输入错误，请重新输入！")
            continue
    break

update_by_phone()

更改学生信息

def update_stu(): stus = read_stu_csv() stu_names = stus['姓名'].values.tolist() stu_phones = stus['电话'].values.tolist() display(stus) input_flag = True while input_flag: stu_input = input("请输入学生姓名/电话：") if stu_input in stu_names: stu = stus.loc[(stus['姓名'] == stu_input),:] input_flag = False break if int(stu_input) in stu_phones: stu = stus.loc[(stus['电话'] == int(stu_input)),:] input_flag = False break else: print_message("未查询到【{}】信息，请重新输入".format(stu_input)) continue

display(stu)

while True:
    thing = input("请输入要修改的类型（'学号ID', '姓名', '电话', '生日', '入学年份'）：")
    if thing not in ["学号ID", "姓名", "电话", "生日", "入学年份"]:
        print_message("类型选择有误，请重新输入")
        continue
    else:
        
        # 获取学生原信息
        stu_dict = stu.to_dict()[thing]
        stu_index = list(stu_dict.keys())[0]
        stu_one = stu_dict[stu_index]
        print_message("原信息为: {}". format(stu_one))
        new_one = input("请输入新内容：")
        print_message("您确定将【{}】修改为【{}】吗？".format(stu_one, new_one))
        confirm_flag = input("y/n：")
        if confirm_flag == "y":
            thing_index = list(stus.columns).index(thing)
            stus.iloc[stu_index, thing_index] = new_one
            # 写入stu表中(stu.csv)
            stus.to_csv("stu.csv", index = False)
            display(stus)
            break

update_stu()