9.python-读取txt文档

数据处理背景：
读取实验后数据记录文件 1_teacher.txt ，只获取文档中的 quality，delay，varQuality。
分别保存成txt文档，绘制成CDF。

import csv


def openreadtxt(file_name):
    quality = []
    delay = []
    varQuality = []

    with open(file_name, 'r') as file_data:
        for row in file_data:
            tmp_list = row.strip("\n").split(',')  # 去除首尾换行符，按‘,'切分每行的数据

            quality.append(tmp_list[1])
            delay.append(tmp_list[4])
            varQuality.append(tmp_list[7])

    return quality, delay, varQuality


def openwritetxt(quality, delay, varQuality):
    file_quality = open('quality.txt', mode='w')
    file_delay = open('delay.txt', mode='w')
    file_varQuality = open('varQuality.txt', mode='w')

    for i in range(len(quality)):
        file_quality.writelines(quality[i] + '\n')
    file_quality.close()

    for i in range(len(delay)):
        file_delay.writelines(delay[i] + '\n')
    file_delay.close()

    for i in range(len(varQuality)):
        file_varQuality.writelines(varQuality[i] + '\n')
    file_varQuality.close()


if __name__ == "__main__":
    quality, delay, varQuality = openreadtxt('1_teacher.txt')
    openwritetxt(quality, delay, varQuality)
    # print(data)