Добавление прогнозируемого вывода в исходный фрейм данных

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)

Приведенное выше создаст фрейм данных predictDf, который будет иметь 2 столбца - Year с прогнозируемыми годами и Prediction с прогнозируемыми значениями.