一篇文章按照======分割成三段
f = open("C:\\Users\\Administrator\\Desktop\\新建文本文档.txt")
boy = []
girl = []
count = 1
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_name_boy = "boy_" + str(count) + ".txt"
file_name_girl = "girl_" + str(count) + ".txt"
boy_file = open(file_name_boy,"w")
girl_file = open(file_name_girl,"w")
boy_file.writelines(boy)
girl_file.writelines(girl)
boy_file.close()
girl_file.close()
boy = []
girl = []
count += 1
file_name_boy = "boy_" + str(count) + ".txt"
file_name_girl = "girl_" + str(count) + ".txt"
boy_file = open(file_name_boy,"w")
girl_file = open(file_name_girl,"w")
boy_file.writelines(boy)
girl_file.writelines(girl)
boy_file.close()
girl_file.close()
f.close()
简化包装后
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,"w")
girl_file = open(file_name_girl,"w")
boy_file.writelines(boy)
girl_file.writelines(girl)
boy_file.close()
girl_file.close()
def split_file(file_name):
f = open("C:\\Users\\Administrator\\Desktop\\新建文本文档.txt")
boy = []
girl = []
count = 1
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:
save_file(boy,girl,count)
boy = []
girl = []
count += 1
save_file(boy,girl,count)
f.close()
split_file("C:\\Users\\Administrator\\Desktop\\新建文本文档.txt")