python调用函数实现数据的增删改查(2)

1  添加数据

def add():
# 输入姓名,年龄,电话
name=raw_input('name:')
age=raw_input('age:')
phone=raw_input('phone:')
#封装个人信息,放入list里面
stu=[name,age,phone]
#把学生放入数据源data_list
data_list.append(stu)
print data_list
print 'hi,welcome to Beijing!'

注意:当添加完数据桌面会新建一个txt文档,添加数据会查看到内容,如下:

2删除数据

def delete():
query()
num=input('请输入要删除的序号:')
while num<0 or num>len(data_list):
num=input('请输入正确的序号:')
del data_list[num-1]
print '删除数据成功!'

3   修改数据,并没有完善好这里

4  查询数据

def query():
count=0
for stu in data_list:
count+=1
print count,'name',stu[0] ,'age:',stu[1],'phone:',stu[2]
print '_'*10

 

5  退出程序

 

具体代码如下:

#coding:utf-8
print '''
1 添加数据
2 删除数据
3 修改数据
4 查看数据
5 退出程序
'''
data_list=[]
#定义加载数据函数load_data
def load_data():
f=open('stu.txt','r') #读取数据
res=f.readlines() #读取多行
for stu in res:
print stu
#把学生放入数据源data_list


def add():
# 输入姓名,年龄,电话
name=raw_input('name:')
age=raw_input('age:')
phone=raw_input('phone:')
#封装个人信息,放入list里面
stu=[name,age,phone]
#把学生放入数据源data_list
data_list.append(stu)
print data_list
print 'hi,welcome to Beijing!'
def delete():
query()
num=input('请输入要删除的序号:')
while num<0 or num>len(data_list):
num=input('请输入正确的序号:')
del data_list[num-1]
print '删除数据成功!'
def update():
print 'hi,welcome to Shanghai!'
def query():
count=0
for stu in data_list:
count+=1
print count,'name',stu[0] ,'age:',stu[1],'phone:',stu[2]
print '_'*10
def exit():
print '退出程序'
#保存数据到txt文档
def save():
f=open('/Users/mac/Desktop/stu1.txt','w')
#遍历数据源,把学生信息写入文档
for stu1 in data_list:
#遍历学生信息
for item in stu1:
f.write(item)
if item==stu1[-1]:
f.write('\n')
print 'save....'
data_list=[]

while 1:
num=input('请选择您的操作:')
if num==1:
add()
# save()
elif num==2:
delete()
elif num==3:
update()
elif num==4:
query()
elif num==5:
exit()
break
else:
print 'error'
save()

本文来自博客园,作者:Jaoany,转载请注明原文链接:https://www.cnblogs.com/fanglijiao/p/6970462.html

原文地址:https://www.cnblogs.com/fanglijiao/p/6970462.html