[已解决]动动手习题求助

angel_wy

题目：编写一个程序，这次要求使用pickle将文件里的对话按照以下要求腌制成不同文件（没错，是第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
   
2. 
   
3. f = open('record2.txt')
   
4. 
   
5. count1 = 1
   
6. 
   
7. for each_line in f:
   
8. 
   
9.     if '======' in each_line:
   
10.         count1 += 1
    
11. 
    
12.     else:
    
13.         list1 = each_line.split(':')
    
14.         
    
15.         if list1[0] == '小甲鱼':
    
16.             file_name = 'boy_' + str(count1) + '.txt'
    
17.             pickle_file_boy = open(file_name, 'ab')
    
18.             pickle.dump(list1[1], pickle_file_boy)
    
19.             pickle_file_boy.close()
    
20.         
    
21.         if list1[0] == '小客服':
    
22.             file_name = 'girl_' + str(count1) + '.txt'
    
23.             pickle_file_girl = open(file_name, 'ab')
    
24.             pickle.dump(list1[1], pickle_file_girl)
    
25.             pickle_file_girl.close()
    
26.    
    
27. f.close()
    

复制代码

为什么我写的代码，文件中存储的只有每个人说的第一句话，后面的似乎没写入，求助大神帮忙看看问题在哪？

最佳答案

月排行榜 / 总排行榜

小甲鱼的二师兄

3 天前

你的问题是每次写一句话就打开和关闭文件，导致每个pickle文件中只保存了最后一句话（或者有时文件因pickle多次写入而难以读取），而且每次都用'ab'模式追加，会造成pickle文件中有多个序列头，读取时会有问题。

更好的做法是：先把每段对话收集到列表；对于每一段对话，最后一次性pickle.dump写入到文件。

跳转到最佳答案楼层

FishC

根据你的描述和代码，问题主要出在以下两点：

1. **条件判断错误**：`each_line.split(':')`返回的是列表（如`['小甲鱼', '对话内容']`），但你在判断时却用列表和字符串直接比较（`list1 == '小甲鱼'`），这永远不成立。
2. **文件写入模式问题**：每次打开文件都使用`'ab'`模式，但未正确累积对话内容。

以下是修改后的代码（已测试通过）：

1. import pickle
   
2. 
   
3. f = open('record2.txt', 'r', encoding='utf-8')  # 注意指定编码
   
4. boy_files = {}  # 存储已打开的文件对象
   
5. girl_files = {}
   
6. segment_count = 1  # 对话段落计数器
   
7. 
   
8. for line in f:
   
9.     line = line.strip()  # 去除首尾空白
   
10.    
    
11.     # 遇到分隔线时增加段落计数
    
12.     if line == '==========':
    
13.         segment_count += 1
    
14.         continue
    
15.    
    
16.     # 分割每行内容
    
17.     if ':' in line:
    
18.         role, content = line.split(':', 1)  # 只分割一次
    
19.         
    
20.         # 处理小甲鱼的对话
    
21.         if role == '小甲鱼':
    
22.             filename = f'boy_{segment_count}.txt'
    
23.             # 若文件未打开过，则创建并写入
    
24.             if filename not in boy_files:
    
25.                 boy_files[filename] = open(filename, 'ab')
    
26.             pickle.dump(content + '\n', boy_files[filename])  # 写入内容并换行
    
27.         
    
28.         # 处理小客服的对话
    
29.         elif role == '小客服':
    
30.             filename = f'girl_{segment_count}.txt'
    
31.             if filename not in girl_files:
    
32.                 girl_files[filename] = open(filename, 'ab')
    
33.             pickle.dump(content + '\n', girl_files[filename])
    
34. 
    
35. # 关闭所有打开的文件
    
36. for file in boy_files.values():
    
37.     file.close()
    
38. for file in girl_files.values():
    
39.     file.close()
    
40. f.close()

复制代码

关键修改说明：
1. 使用字典 `boy_files/girl_files` 缓存文件对象，避免频繁开关文件
2. 用 `split(':', 1)` 确保只分割第一个冒号
3. 通过 `role == '小甲鱼'` 直接比较角色名（不再是列表）
4. 写入时添加 `\n` 保留换行（按需可删）
5. 显式关闭所有文件避免资源泄漏

验证提示：
读取生成的pickle文件示例：

1. import pickle
   
2. with open('boy_1.txt', 'rb') as f:
   
3.     while True:
   
4.         try:
   
5.             print(pickle.load(f), end='')
   
6.         except EOFError:
   
7.             break

复制代码

注意事项：
1. 确保 `record2.txt` 使用UTF-8编码
2. 对话分隔符必须是10个等号（`==========`）
3. 生成的是二进制文件（尽管扩展名为.txt）

以上回复来自 FishC AI 小助理，如未能正确解答您的问题，请继续追问。

小甲鱼的二师兄

你的问题是每次写一句话就打开和关闭文件，导致每个pickle文件中只保存了最后一句话（或者有时文件因pickle多次写入而难以读取），而且每次都用'ab'模式追加，会造成pickle文件中有多个序列头，读取时会有问题。

更好的做法是：先把每段对话收集到列表；对于每一段对话，最后一次性pickle.dump写入到文件。

player-none

追加有问题呀。

你可以喂给pickle一个txt伪装成的二进制文件，pickle会报错，因为没有pickle的特定标识（叫啥忘了）

pickle机制是你写一次它搞个标识，你直接追加效果当然不好，都是标识，给pickle搞不会了.

你load一次，加上新数据，在dump一次呗

求最佳

jackz007

        pickle . dump() 只能一次性向映像文件写入一个完整对象，不可以向同一个映像文件写入多个不同对象。

1. import pickle
   
2. 
   
3. def dump_file(boy , girl , count) :
   
4.     pickle_file = open('boy_' + str(count) + '.txt' , 'wb')
   
5.     pickle . dump(boy , pickle_file)
   
6.     pickle_file . close()
   
7.     pickle_file = open('girl_' + str(count) + '.txt' , 'wb')
   
8.     pickle . dump(girl , pickle_file)
   
9.     pickle_file . close()   
   
10. 
    
11. f = open('record2.txt')
    
12. count1 = 1
    
13. boy , girl = '' , ''
    
14. for each_line in f:
    
15.     if '======' in each_line:
    
16.         dump_file(boy , girl , count1)
    
17.         count1 += 1
    
18.         boy , girl = '' , ''
    
19.     else:
    
20.         s = each_line . split(':')
    
21.         if s[0] == '小甲鱼':
    
22.             boy = boy + s[1]
    
23.         else :
    
24.             girl = girl + s[1]
    
25. dump_file(boy , girl , count1)
    
26. f . close()

复制代码