将预测输出添加到原始数据帧

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.frame predictDf，其中包含 2 列 - Year 包含要预测的年份，Prediction 包含预测值。

将预测输出添加到原始数据帧

Adding predict output to original dataframe

r

predict