文件-一个简单的任务

最新推荐文章于 2023-02-03 16:00:16 发布

原创最新推荐文章于 2023-02-03 16:00:16 发布 · 229 阅读

0 ·

CC 4.0 BY-SA版权

Python 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了一种从单一记录文件中分离对话的方法，通过Python脚本实现，可以将不同角色的对话内容分割并保存到独立的文本文件中。适用于需要对文本对话进行分类处理的场景。

2016.12.16

一般文件分为输入，处理和输出三个部分。

下面是有关文件的一张表。

如何打开文件

handle = open(file_name,access_mode='r')

file_name变量包含我们希望打开的文件的字符串名字，access_mode中‘r’表示读取，‘w’表示写入，‘a’表示添加。‘+’表示读写，‘b’表示二进制访问。如果未提供access_mode，默认值为‘r’.如果open成功，一个文件对象句柄会被返回。当一个文件被返回之后，我们就可以访问它的一些方法，比如readlines()和close().

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

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

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

–文件中总共有三段对话，分别保存为boy_1.txt, girl_1.txt，boy_2.txt, girl_2.txt, boy_3.txt, gril_3.txt共6个文件（提示：文件中不同的对话间已经使用“==========”分割）

关于一个文件，编写一个程序如下：

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

运行后能够在同一个文件中生成几个txt形式的文件。注意代码生成的地方要和文件的地方在一起。这样就可以打开和读取文件了。