第一次用python(pychram) 写代码管理文件（打开、写入、关闭）(简单函数调用)

转载于 2018-07-08 14:31:44 发布 · 652 阅读

文章标签：

#python #文件管理

程序人生专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一个简单的Python程序，用于将聊天记录文件按角色分为男生和女生两个独立的文本文件。程序首先读取原始记录文件，然后根据对话内容中的角色标签进行分类，并将分类后的对话保存到对应的文件中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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')