python基础第31讲pickle模块动动手

yoyo2

1. '''
   
2. 编写一个程序，这次要求使用pickle将record.txt文件里的对话按照以下要求
   
3. 腌制成不同文件（没错，是第29讲的内容小改，考考你自己能写出来吗？）：
   
4. 
   
5.     小甲鱼的对话单独保存为boy_*.txt的文件（去掉“小甲鱼:”）
   
6.     小客服的对话单独保存为girl_*.txt的文件（去掉“小客服:”）
   
7.     文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt，boy_2.txt, girl_2.txt, boy_3.txt, gril_3.txt共6个文件（提示：文件中不同的对话间已经使用“==========”分割）
   
8. '''
   
9. import pickle
   
10. import os
    
11. 
    
12. os.chdir('D:\\workspace')
    
13. #一次读一行
    
14. #如果内容不是=========
    
15. #-----以“ ：”分离，如果前面是小甲鱼就放boy[]，如果前面是小客服就放[]
    
16. #如果内容是=========
    
17. #-----保存数据;count += 1;boy[]和girl[]清零
    
18. #因为只有两行========，所以要还有执行保存数据一次
    
19. def save(boy,girl,count):
    
20.     boy_file = 'boy_' + str(count) + '.txt'#文件名
    
21.     girl_file = 'girl_' + str(count) + '.txt'#文件名
    
22. 
    
23.     file_save = open(boy_file,'wb')
    
24.     pickle.dump(boy,file_save)
    
25.     file_save.close()
    
26. 
    
27.     file_save = open(girl_file,'wb')
    
28.     pickle.dump(girl,file_save)
    
29.     file_save.close()
    
30. 
    
31. def split_file(file):
    
32.     count = 1
    
33.     boy = []
    
34.     girl = []
    
35.    
    
36.     file = open(file)
    
37. 
    
38.     for each_line in file:
    
39.         if each_line[:6] != '======': #报错处1,原代码为if each_line != '========='会无法判断========行
    
40.             (person, talk) = each_line.split(':',1)#报错处2，原代码为(person, talk) = each_line.split(':')原因是有一行有两个“ ：”系统无法判断怎么分割
    
41.             if person == '小甲鱼':
    
42.                 boy.append(talk)
    
43.             elif person == '小客服':
    
44.                 girl.append(talk)
    
45.         else:
    
46.             save(boy,girl,count)
    
47. 
    
48.             boy.clear()
    
49.             girl.clear()
    
50. 
    
51.             count += 1
    
52.     save(boy,girl,count)#漏这步只能产生两个文件，因为只有两行“=========”
    
53. 
    
54. split_file('D:\\workspace\\record.txt')

复制代码

Twilight6

这里的报错处 1 和 2 实际上应该只有一个

这里  if each_line != '=========' 应该不会报错，而报错原因都应该是第二个

这里是因为当你直接 if each_line != '=========' 时候，Python 按行读取txt 文本时候会自带 \n 换行符，所以 if 条件永远都会成立

导致 split 切割这行都是等号的时候，因为没有冒号，所以 split 只返回含有一个元素的列表，而你设置了两个变量接受切割后的列表元素，所以导致的报错

当你代码改成 (person, talk) = each_line.split(':') 同理，并不是因为系统无法判断怎么分隔，是因为系统无法分配切割后的元素

yoyo2

Twilight6 发表于 2020-8-23 19:24
这里的报错处 1 和 2 实际上应该只有一个

这里  if each_line != '=========' 应该不会报错，而报错 ...

噢，原来是这样，多谢，我split理解错了