diff --git a/lib/tools.py b/lib/tools.py index 6c8264b..b3548b7 100644 --- a/lib/tools.py +++ b/lib/tools.py @@ -1114,6 +1114,12 @@ def plot_pp_predict_result(y_hat, global_config, wd='yuedu'): if wd == 'yuedu': y = pd.read_csv(os.path.join( global_config['dataset'], '指标数据.csv'))[['ds', 'y']][-12:] + print('月度历史数据日期更改') + y['ds'] = pd.to_datetime(y['ds']) + y['ds'] = y['ds'].dt.strftime('%Y-%m-01') + # ds 转换为日期类型 + y['ds'] = pd.to_datetime(y['ds']) + # 修改ds列的日为1日 print(y_hat) y_hat['ds'] = pd.to_datetime(y_hat['ds']) @@ -1155,6 +1161,9 @@ def plot_pp_predict_result(y_hat, global_config, wd='yuedu'): y_y_hat = y_y_hat.sort_values(by='ds') y = y.sort_values(by='ds') + # # y的日期转为1日 + # y['ds'] = y['ds'].dt.strftime('%Y-%m-01') + # 绘制 y_hat 的折线图,颜色为橙色 sns.lineplot(x=y_y_hat['ds'], y=y_y_hat['predictresult'], color='orange', label='预测值', ax=ax, linestyle='--', linewidth=2)