ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

通过python读取txt文件和绘制柱形图

2021-03-13 23:00:23  阅读:212  来源: 互联网

标签:span python 柱形图 label content value time txt


目录

目的

使用效果

代码实现


目的

临床数据的记录时间对应标签(逗号后面的数字)记录在txt文件里,要把标签转换为3类标签,并且计算出每个标签的分别持续时间,然后绘制成柱形图方便查阅。

小难点分析:(1)txt的切割读取对应内容   (2)时间差计算

txt文件如图:

使用效果

首先将原始txt转换为  左列新标签 右列持续时间

绘制为柱形图

为了直观,每次只最多显示 2个小时,同时横坐标还是按照临床的记录时间顺序。

代码实现

# -*- coding: utf-8 -*-

from datetime import datetime
import matplotlib.pyplot as plt
from matplotlib.font_manager import FontProperties

font_set = FontProperties(fname=r"c:\windows\fonts\simsun.ttc", size=12)
address = "D:/XXXXXX/"
patient = "XXXXX"
year = '2020'  # or 2020
txt_address = address + patient + "/timetable.txt"
filename = address + patient + "/newTimeTable.txt"  # 保存的txt地址

# 读取txt的相应内容
def showLabel():
    content = []
    label = []
    rawtime = []

    with open(txt_address, 'r', encoding="utf-8") as f:
        lines = f.readlines()  # 返回的是列表

    print("文件的行数:%d" % len(lines))

    for line in lines:
        content_value = line.split(sep=',')  # 分为[0]和[1]
        time_value = content_value[0].split(sep=year)
        content_value = content_value[1].split(sep='\n')
        temp_time = year + time_value[-1]
        rawtime.append(temp_time)
        content.append(content_value[0])

    # 去掉第一行的line和转换label
    for i in range(1, len(lines)):  # range是左闭右开
        number = int(content[i])
        if number == 4:
            label_value = 1  # label 1 表示癫痫发作
        elif number == 1:
            label_value = 0  # label 0 表示数据缺失
        else:
            label_value = -1  # label -1 表示非癫痫发作
        label.append(label_value)

    del rawtime[0]  # 去掉第一行的line
    length = len(lines)
    return label, rawtime, length

# 计算时间差
def getTimeSpan(new_label,time,length):
    time_span = []
    time_label = []
    temp = datetime.strptime(time[0], "%Y-%m-%d    %H:%M:%S")
    for i in range(0, length - 2):
        if new_label[i] != new_label[i + 1]:
            end_time = datetime.strptime(time[i + 1], "%Y-%m-%d    %H:%M:%S")
            diff = end_time - temp
            temp = end_time
            # time_span.append(diff) 标准化时间
            time_span.append(round(diff.total_seconds() / 60, 2))  # 保留两位小数
            time_label.append(new_label[i])

    return time_span, time_label

# 画图,按照标签内容给图上色
def showTimeandLabel(time_span,time_label):
    Yanse = []
    for i in time_label:
        if i == 0:
            yanse = 'blue'
        elif i == -1:
            yanse = 'gray'
        else:
            yanse = 'red'
        Yanse.append(yanse)

    plt.bar(range(len(time_span)), time_span, width=0.8, tick_label=time_label, color=Yanse)
    plt.xlabel('标签', fontproperties=font_set)
    plt.ylabel("持续时间(分钟)", fontproperties=font_set)
    plt.title("patient: " + patient, size=20)

    plt.ylim(0, 120)
    plt.show()

# 保存到txt
def txtSave(time_span, time_label):
    # w:向文件中写入内容时,会先清空原文件中的内容,
    with open(filename, 'w', encoding="utf-8") as f:
        line_number = len(time_label)
        for i in range(line_number):
            f.write(str(time_label[i])+"                           "+str(time_span[i]))
            f.write("\n")


def main():
    new_label, time, length = showLabel()
    time_span, time_label = getTimeSpan(new_label, time, length)
    txtSave(time_span, time_label)
    showTimeandLabel(time_span, time_label)


if __name__ == '__main__':
    main()

 

标签:span,python,柱形图,label,content,value,time,txt
来源: https://blog.csdn.net/Amigo_1997/article/details/114766182

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有