任务:
将文件(record.txt)中的数据进行分割并按照以下规律保存起来
-小甲鱼的对话单独保存为boy_*.txt(去掉“小甲鱼:”)
-小客服的对话单独保存为girl_*.txt的文件(去掉“小客服:”)
-文件中总共有三段对话,分别保存为boy_1.txt,girl_1.txt,boy_2,txt,girl_2.txt,boy_3.txt,girl_3.txt共6个文件(提示:文件中不同的对话间已经使用“==========”分割)
代码及注释:
def save_file(boy,girl,count): #定义一个建立boy和girl文件的函数
file_name_boy = "boy_" + str(count) + ".txt" #定义boy的文件名
file_name_girl = "girl_" + str(count) + ".txt"#定义girl的文件名
boy_file = open(file_name_boy,"w") #以写入方式打开boy文件
girl_file = open(file_name_girl,"w")#以写入方式打开girl文件
boy_file.writelines(boy)#将boy列表中的每一行都写入boy_file
gilr_file.writelines(girl)#将girl列表中的每一行都写入girl_file
boy_file.close()#关闭boy文件
girl_file.close()#关闭girl文件
def split_file(file_name):#定义函数
f = open("record.txt")#打开文件
boy = []#建立命名为boy的空列表
girl = []#建立命名为girl的空列表
count = 1#给count赋值为1
for each_line in f:#用for循环读取文件中的每一行
if each_line[:6] != "=======":#判断读取的那一行是否为分割线
(role,line_spoken) = each_line.split(":",1)#将读取的那一行以“:”分片为两部分,前面那部分为role,后面那部分为line_spoken
if role == "小甲鱼":
boy.append(line_spoken)#如果role为小甲鱼的话,就将这一行的line_spoken写入boy这一列表
if role == "小客服":
girl.append(line_spoken)#如果role为小客服的话,就将这一行的line_spoken写入girl这一列表
else:
save_file(boy,girl,count)#如果读取的那一行是分割线,则运行save_file函数
boy = []#使boy列表重新定义为空列表
girl = []#使girl列表重新定义为空列表
count += 1#使count加1
save_file(boy,girl,count)#调用save_file函数,将第三段对话也分别写入boy_3.txt和girl_3.txt文件中
f.close()#关闭f文件
split_file("record.txt")#调用split_file函数
(注来源:《[小甲鱼]零基础入门学习Python》)