课时29：文件：一个任务

Posted 2020-12-31 那是个好男孩

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了课时29：文件：一个任务相关的知识，希望对你有一定的参考价值。

目录：

　　一、一个任务

　　二、课时29课后习题及答案

****************

一、一个任务

****************

任务：将文件（record.txt）中的数据进行分割并按照以下规律保存起来：

（1）小甲鱼的对话单独保存为boy_*.txt的文件（去掉“小甲鱼:”）

（2）小客服的对话单独保存为girl_*.txt的文件（去掉“小客服:”）

（3）文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt，boy_2.txt, girl_2.txt, boy_3.txt, gril_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\')