文件读写三部曲:打开文件–>读/写文件–>关闭文件。下面通过实例来讲解pickle模块是如何完成文件读写工作。
pickle提供了一个简单的持久化功能。python中几乎所有的数据类型(列表,字典,集合,类等)都可以用pickle来序列化。
(1)使用pickle模块写入数据:
Step1:pickle_file =open(‘my_data.pkl’,’wb’) # 以二进制写入模式打开名为my_data.pkl的文件;
Step2:pickle.dump(data,pickle_file) #把data中的内容写入my_data.pkl,pickle_file为Step1打开的文件对象;
Step3:pickle_file.close()#关闭文件。
(2)读取pickle腌制的文件:
Step1:pickle_file =open(‘(‘my_data.pkl’,’rb’) #以二进制读取模式打开文件my_list.pkl
Step2:temp_data =pickle.load(pickle_file) #读取pickle_file中的内容,并保存到temp_data中
Step3:·pickle_file.close()#关闭文件
课后习题及答案:
0.pickle的实质是什么?
使用一些算法将数据对象“腌制”成二进制形式,存储在磁盘上,当然也可以放在数据库或者通过网络传输到另一台计算机上。
1.使用pickle的什么方法存储数据?
使用pickle.dump(data,obj_file)。其中,data是待存储的数据对象,obj_file是目标存储的文件对象,该文件一定要先使用’wb’的模式打开,否则会报错。
2.使用pickle的什么方法读取数据?
使用pickle.load(obj_file),obj_file是目标存储的文件对象,该文件一定要使用’rb’的模式打开。
3.使用pickle能不能保存为”*.txt”类型的文件?
可以。也可以保存为‘.pkl’或者其他你想保存的格式。
4.编写一个程序,这次要求使用pickle将文件(record.txt)里的对话按照以下要求腌制成不同文件(没错,是第29讲的内容小改,考考你自己能写出来吗?):小甲鱼的对话单独保存为boy_*.txt的文件(去掉“小甲鱼:”)小客服的对话单独保存为girl_*.txt的文件(去掉“小客服:”)文件中总共有三段对话,分别保存为boy_1.txt,girl_1.txt,boy_2.txt, girl_2.txt, boy_3.txt,gril_3.txt共6个文件(提示:文件中不同的对话间已经使用“==========”分割)
代码实现:
## pickle_file.py
import pickle
def save_pkl_file(boy, girl, count):
boy_name ='boy_' + str(count) + '.pkl'
girl_name ='girl_' + str(count) + '.pkl'
file_boy =open(boy_name, 'wb')
file_girl =open(girl_name, 'wb')
pickle.dump(boy, file_boy, True)
pickle.dump(girl,file_girl, True)
file_boy.close()
file_girl.close()
def split_file(filename):
f =open(filename, 'r')
count = 1
boy = []
girl = []
foreach_line in f:
ifeach_line[:6] !='======':
(role, spoken) = each_line.split(':',1)
ifrole == '小甲鱼':
boy.append(each_line)
elifrole == '小客服':
girl.append(each_line)
else:
save_pkl_file(boy, girl, count)
count+=1
boy= []
girl= []
save_pkl_file(boy, girl, count)
filename =r'C:\Users\XXN\Desktop\python\fishC_31\record.txt'
split_file(filename)
运行结果: