pickle文件处理

gzj137070928

# 将record文件里面的内容按照文中的“======”进行分割，并按照下面的要求使用pickle保存：
# 小甲鱼的对话单独保存为boy_*.txt的文件，并去掉“小甲鱼：”
# 小客服的对话单独保存为girl_*.txt的文件，并去掉“小客服：”
# 文中有两处“======”，那么对应会产生boy_1.txt、boy_2.txt、boy_3.txt和
# girl_1.txt、girl_2.txt、girl_3.txt这6个文件

1. import pickle
   
2. 
   
3. 
   
4. def file_write(listname, role, count):
   
5.     filename = role + str(count) + '.txt'
   
6.     f = open(filename, 'wb')
   
7.     for i in listname:
   
8.         pickle.dump(i, f)
   
9.     f.close()
   
10. 
    
11. 
    
12. def file_split(filename):
    
13.     f = open(filename, 'r', encoding='utf-8')
    
14.     count =1
    
15.     boys = []
    
16.     girls = []
    
17.     for eachline in f:
    
18.         if not('===' in eachline):
    
19.             (role, speak) = eachline.split('：', 1)
    
20.             if role == '小甲鱼':
    
21.                 boys.append(speak)
    
22.             elif role == '小客服':
    
23.                 girls.append(speak)
    
24.         else:
    
25.             file_write(boys, 'boy', count)
    
26.             file_write(girls, 'girl', count)
    
27.             count += 1
    
28.             boys = []
    
29.             girls = []
    
30.     file_write(boys, 'boy', count)
    
31.     file_write(girls, 'girl', count)
    
32.     f.close()
    
33. 
    
34. 
    
35. file_split('record.txt')
    

复制代码