Question

我使用由Google人员开发的CausalImpact软件包来评估干预措施在时间序列中的因果影响。

我创建了一个情节，就像在示例中一样。我想在第249页添加一个图例，就像我们找到here一样。

我知道使用函数图（CausalImpact包）会产生一个ggplot2对象。但是，我不知道如何添加图例。

一些可重现的数据：

library(devtools)
devtools::install_github("google/CausalImpact")
library(CausalImpact)

set.seed(1)
x1 <- 100 + arima.sim(model = list(ar = 0.999), n = 100)
y <- 1.2 * x1 + rnorm(100)
y[71:100] <- y[71:100] + 10
data <- cbind(y, x1)
pre.period <- as.Date(c("2014-01-01", "2014-03-11"))
post.period <- as.Date(c("2014-03-12", "2014-04-10"))
time.points <- seq.Date(as.Date("2014-01-01"), by = 1, length.out = 100)
data <- zoo(cbind(y, x1), time.points)
impact <- CausalImpact(data, pre.period, post.period)
plot(impact)

Answer 1

为了添加更多元素（我认为这是你的想法），你需要调整功能：

CausalImpact:::CreateImpactPlot

有关源代码，请参阅here。

Answer 2

或者，您可以使用impact[["series"]]

中的数据创建自己的地图

如何将图例添加到CausalImpact绘图对象

2 个答案: