在 R 的预测包中添加预测线的图例条目

Question

我有一个通过fcast <- forecast(Arima(data$train, order=...))获得的预报对象图：

fcast %>% autoplot(series='Forecast') +
  autolayer(fitted(fit), series='Fitted') + 
  autolayer(data$train, series = 'Train') +
  autolayer(data$test, series='Test') +
  xlab("Observation [days]") +
  ylab("Energy [Watts]") +
  guides(colour=guide_legend(title="Data series"), 
         fill=guide_legend(title="Prediction interval"))

看起来像这样：

如您所见，除了代表预测点的深蓝色线外，我将所有内容都添加到了图例中。

如何将预测点（深蓝色线）添加到图例中？

编辑：解决方案，基于 Rob Hyndman 的回答：

clrs <- c("blueviolet", "blue", "darkgoldenrod4", "red")

autoplot(fcast) +
  autolayer(fcast$mean, series="Forecast") +
  autolayer(fitted(fit), series='Fitted') + 
  autolayer(data$train, series = 'Train') +
  autolayer(data$test, series='Test') +
  xlab("Observation [days]") +
  ylab("Energy [Watts]") +
  guides(colour=guide_legend(title="Data series"), 
         fill=guide_legend(title="Prediction interval")) +
  scale_color_manual(values=clrs)

Answer 1

再添加一次 autolayer:

autoplot(fcast) +
  autolayer(fcast$mean, series="Forecast") +
  autolayer(fitted(fit), series='Fitted') + 
  autolayer(data$train, series = 'Train') +
  autolayer(data$test, series='Test') +
  xlab("Observation [days]") +
  ylab("Energy [Watts]") +
  guides(colour=guide_legend(title="Data series"), 
         fill=guide_legend(title="Prediction interval"))

那么你可能需要玩一下颜色。

Answer 2

我做了以下，但还是无法得到图例

 autoplot(arima_kw_Pat)+
      xlab("Day")+
      ylab("Under Treatment")+
      ggtitle("Patients Under Treatment in Kuwait ARIMA")+
      guides(colour=guide_legend(title="Data series"), 
              fill=guide_legend(title="Prediction interval"))+
      scale_color_manual(values=clrs)

相反，我收到了以下错误 is_missing(值) 中的错误：找不到对象 'clrs'

在 R 的预测包中添加预测线的图例条目

Add legend entry for prediction line in R's forecast package

r

ggplot2

forecasting