python 绘图 异常点绘制使用 ax.plot(abnormal_points['ds'], abnormal_points['y'], "rX", label='abnormal points')
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 | from matplotlib import pyplot as plt def my_plot(title, m, fcst, ax = None , uncertainty = True , plot_cap = True , xlabel = 'ds' , ylabel = 'y' , abnormal_points = None ): """Plot the Prophet forecast. Parameters ---------- m: Prophet model. fcst: pd.DataFrame output of m.predict. ax: Optional matplotlib axes on which to plot. uncertainty: Optional boolean to plot uncertainty intervals. plot_cap: Optional boolean indicating if the capacity should be shown in the figure, if available. xlabel: Optional label name on X-axis ylabel: Optional label name on Y-axis Returns ------- A matplotlib figure. """ if ax is None : fig = plt.figure(facecolor = 'w' , figsize = ( 10 , 6 )) ax = fig.add_subplot( 111 ) else : fig = ax.get_figure() fcst_t = fcst[ 'ds' ].dt.to_pydatetime() ax.plot(m.history[ 'ds' ].dt.to_pydatetime(), m.history[ 'y' ], 'k.' , label = 'y' ) ax.legend() ax.plot(fcst_t, fcst[ 'yhat' ], ls = '-' , c = '#0072B2' , label = 'predicted y' ) ax.legend() ax.fill_between(fcst_t, 0 , fcst[ 'yhat_upper' ], color = '#0072B2' , alpha = 0.2 , label = 'predicted upper y' ) # ax.plot(fcst_t, fcst['yhat_upper'], ls='--', color='#0072B2', alpha=0.2, label='predicted upper y') ax.legend() if abnormal_points is not None : ax.plot(abnormal_points[ 'ds' ], abnormal_points[ 'y' ], "rX" , label = 'abnormal points' ) ax.legend() ax.set_title(title) ax.grid( True , which = 'major' , c = 'gray' , ls = '-' , lw = 1 , alpha = 0.2 ) ax.set_xlabel(xlabel) ax.set_ylabel(ylabel) fig.tight_layout() plt.savefig( "png/{}.png" . format (title)) return fig |
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 记一次.NET内存居高不下排查解决与启示
· 探究高空视频全景AR技术的实现原理
· 理解Rust引用及其生命周期标识(上)
· 浏览器原生「磁吸」效果!Anchor Positioning 锚点定位神器解析
· 没有源码,如何修改代码逻辑?
· 全程不用写代码,我用AI程序员写了一个飞机大战
· MongoDB 8.0这个新功能碉堡了,比商业数据库还牛
· 记一次.NET内存居高不下排查解决与启示
· 白话解读 Dapr 1.15:你的「微服务管家」又秀新绝活了
· DeepSeek 开源周回顾「GitHub 热点速览」
2017-09-26 pandas groupby 分组操作