# -*- 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()