You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

75 lines
2.2 KiB

import pandas
print("请选择如下功能:")
print("1.添加学员:")
print("2.删除学员:")
print("3.修改学员信息:")
print("4.查询学员信息:")
print("5.查询所有学员信息:")
print("6.保存学员信息:")
print("7.退出:")
m=0
name_list=[]
xu_list=[]
sex_list=[]
phone_list=[]
while True:
print("请输入你需要的功能序号:")
choice = input()
if choice=='1':
m = m + 1
xu_list.append(m)
print("请输入你的姓名:",end='')
name = input()
name_list.append(name)
print("请输入你的性别:",end='')
sex = input()
sex_list.append(sex)
print("请输入你的电话:",end='')
phone = input()
phone_list.append(phone)
dict1 = {'姓名': name_list, '性别': sex_list, '电话': phone_list}
dataFrame = pandas.DataFrame(dict1, index=xu_list)
# print(dataFrame)
if choice=='2':
print('请输入需要删除学生姓名:',end='')
name=input()
index_name=dataFrame[dataFrame['姓名']==name].index
dataFrame.drop(index_name,inplace=True)
if choice=='3':
print('请输入需要修改学生姓名:',end='')
name=input()
n=name_list.index(name)
print('a.修改姓名:')
print('b.修改性别:')
print('c.修改电话:')
print('d.退出:')
while True:
print('请输入需要修改的信息:')
alt=input()
if alt=='a':
print('输入修改后的姓名:',end='')
name1=input()
name_list[n]=name1
if alt=='b':
print('输入修改后的姓别:', end='')
sex1 = input()
sex_list[n] = sex1
if alt=='c':
print('输入修改后的电话:', end='')
phone1 = input()
phone_list[n] = phone1
if alt=='d':
break
if choice=='4':
print('请输入需要查询的学生姓名:',end='')
name=input()
n = name_list.index(name)+1
print(dataFrame.loc[n])
if choice=='5':
print(dataFrame)
if choice=='7':
break