论.技术.Python.(旧)零学Python第031讲:用 pickle 来保存文件

小蜂队

0.编写一个程序，这次要求使用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个文件（提示：文件中不同的对话间已经使用“==========”分割）
代码如下：

1. import pickle         #导入 pickle 模块
   
2. 
   
3. def save_file(boy, girl, count):
   
4.     file_name_boy = 'boy_' + str(count) + '.txt'    #创建文件名称
   
5.     file_name_girl = 'girl_' + str(count) + '.txt'     #创建文件名称
   
6. 
   
7.     boy_file = open(file_name_boy, 'wb') # 记得一定要加 b 吖   以二进制可以写入的方式打开
   
8.     girl_file = open(file_name_girl, 'wb') # 记得一定要加 b 吖
   
9. 
   
10.     pickle.dump(boy, boy_file)        #用 pickle 保存起来；需要保存的数据是 boy,保存的地址是 boy_file
    
11.     pickle.dump(girl, girl_file)
    
12. 
    
13.     boy_file.close()          #关闭文件
    
14.     girl_file.close()
    
15. 
    
16. def split_file(file_name):
    
17.     count = 1
    
18.     boy = []
    
19.     girl = []
    
20. 
    
21.     f = open(file_name)
    
22. 
    
23.     for each_line in f:
    
24.         if each_line[:6] != '======':
    
25.             (role, line_spoken) = each_line.split(':', 1)   #split() 分割文件这里表示用":" 来分割，分割次数为 1 次
    
26.             if role == '小甲鱼':
    
27.                 boy.append(line_spoken)
    
28.             if role == '小客服':
    
29.                 girl.append(line_spoken)
    
30.         else:
    
31.             save_file(boy, girl, count)
    
32. 
    
33.             boy = []
    
34.             girl = []
    
35.             count += 1
    
36. 
    
37.     save_file(boy, girl, count)
    
38.     f.close()
    
39. 
    
40. split_file('record.txt')
    
41.             

复制代码