微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

Python:使用 matplotlib 的时间序列类可视化

如何解决Python:使用 matplotlib 的时间序列类可视化

我有一个包含时间序列预测类的数组,格式为:

[[DateTime,"Class1"],[DateTime,"Class2"] .... ] 

我想在图中显示原始时间序列数据,下面有一个颜色条,指示每个时间间隔的预测类别,可以看到此颜色条的示例here(x-这里的轴应该是日期时间格式)。

我通过为每个 [DateTime,"Class1"] 对附加像素来实现该示例。但是,问题在于每个连续的 DateTime 之间的间隔是不同的。因此,每个“像素”不应具有相同的宽度。

我的下一个想法是使用线集合,使用具有适当颜色的水平线连接每个日期时间:

def plot_prediction_colorbar(self,ax): 
        classes = self.df["Prediction"].unique() #get all classes
        colors = cm.rainbow(np.linspace(0,1,len(classes))) #color for each one

        dt_pred_arr = self.df[["DateTime","Prediction"]].to_numpy()

        color_arr = []
        line_arr = [] 
        last_coord = None
        last_color = None

        for idx,(dt,pred) in enumerate(dt_pred_arr): 
            dt = date2num(dt)
            cur_coord = [dt,0] 
            if last_coord is not None and last_color is not None:
                color_arr.append(last_color)
                line_arr.append([ last_coord,cur_coord])

            last_coord = cur_coord
            if pred in classes: #For predicted classes
                last_color = colors[np.where(classes==pred)[0]][0]
                
        linecoll = LineCollection(line_arr,colors=color_arr)
        ax.add_collection(linecoll)

但是,没有显示任何行...运行时 line_array 条目的示例是: [[18550.766805555555,0],[18550.766793981482,0]] 这可能是由于日期时间转换导致高浮动而差异很小吗?

任何帮助将不胜感激!

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。