029文件：一个任务

人若有情死得早

任务：将文件（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个文件（提示：文件中不同的对话间已经使用“======”分割）

f = open('record.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('record.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('record.txt')