论.技术.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个文件（提示：文件中不同的对话间已经使用“==========”分割）
代码如下：
import pickle #导入 pickle 模块
def save_file(boy, girl, count):
file_name_boy = 'boy_' + str(count) + '.txt' #创建文件名称
file_name_girl = 'girl_' + str(count) + '.txt' #创建文件名称
boy_file = open(file_name_boy, 'wb') # 记得一定要加 b 吖 以二进制可以写入的方式打开
girl_file = open(file_name_girl, 'wb') # 记得一定要加 b 吖
pickle.dump(boy, boy_file) #用 pickle 保存起来；需要保存的数据是 boy,保存的地址是 boy_file
pickle.dump(girl, girl_file)
boy_file.close() #关闭文件
girl_file.close()
def split_file(file_name):
count = 1
boy = []
girl = []
f = open(file_name)
for each_line in f:
if each_line[:6] != '======':
(role, line_spoken) = each_line.split(':', 1) #split() 分割文件这里表示用":" 来分割，分割次数为 1 次
if role == '小甲鱼':
boy.append(line_spoken)
if role == '小客服':
girl.append(line_spoken)
else:
save_file(boy, girl, count)
boy = []
girl = []
count += 1
save_file(boy, girl, count)
f.close()
split_file('record.txt')
页:
[1]