r语言 - 将预测输出添加到原始数据帧

Question

我有一个数据帧，它有两列 - 年份和录取计数。我有以下代码来预测未来的数字(未来 5 年) (数据框中的最后一年是 1983 年)

df = read.table("C:AdmCounts.txt",header=TRUE)
colnames(df)[1] <- "Year" 
str(df)
model <- lm(AdmCount ~ Year, df)
new.df <- data.frame(Year=c(1985, 1986, 1987, 1988, 1989))
p = predict(model, new.df)

输出为

1        2        3        4        5 
62.64909 62.74266 62.83623 62.92980 63.02337

如何将上述输出添加回原始数据帧以绘制图形？输出中的 1 是 1985 年，2 是 1986 年，依此类推。

谢谢。

Answer 1

您没有指定AdmCount，因此可以自由选择最简单的：

df <- data.frame(
Year = 1985:1989,
AdmCount = 1:5
)
model <- lm(AdmCount ~ Year, df)
predictDf <- data.frame(Year = 1990:1995)
predictDf$Prediction <- predict(model, predictDf)

上面将创建一个 data.framepredictDf，该将有 2 列 -Year列包含要预测的年份，Prediction包含预测值。