00
编写一个程序,这次要求使用pickle将文件(record.txt)里的对话按照以下要求腌制成不同文件:
小甲鱼的对话单独保存为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
def file_save(boy, girl, count):
file_name_boy = ‘boy_‘ + str(count) + ‘.txt‘
file_name_girl = ‘girl_‘ + str(count) + ‘.txt‘
boy_file = open(file_name_boy, ‘wb‘)
girl_file = open(file_name_girl, ‘wb‘)
pickle.dump(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)
if role == ‘小甲鱼‘:
boy.append(line_spoken)
if role == ‘小客服‘:
girl.append(line_spoken)
else:
file_save(boy, girl, count)
boy = []
girl = []
count += 1
file_save(boy, girl, count)
f.close()
split_file(‘record.txt‘)