python录入信息用什么函数_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()