Plotting forecast() objects in ggplot part 1: Extracting the Data

March 13, 2012

(This article was first published on Frank Davenport's Blog on R, Statistics, and all Things Spatial - R, and kindly contributed to R-bloggers)

Lately I’ve been using Rob J Hyndman‘s excellent forecast package. The package comes with some built in plotting functions but I found I wanted to customize and make my own plots in ggplot. In order to do that, I need a generalizable function that will extract all the data I want (forecasts, fitted values, training data, actual observations in the forecast period, confidence intervals, et cetera) and place it into a data.frame with a properly formatted date field (ie, not  a ts() object).

The function below does all that and should work for any forecast object (though I’ve only tested it on Arima() outputs). The only arguments it takes are the original observations and the forecast object (whatever results from calling forecast()). In my next post I’ll give some examples of plotting the results using ggplot and explain why I wanted more than the default plot.forecast() function.


#--Produces a data.frame with the Source Data+Training Data, Fitted Values+Forecast Values, forecast data Confidence Intervals
	require(zoo) #needed for the 'as.yearmon()' function
	en<-max(time(fcast$mean)) #extract the max date used in the forecast
	#Extract Source and Training Data
	#Extract the Fitted Values (need to figure out how to grab confidence intervals)
	ds<-merge(ds,dfit,all.x=T) #Merge fitted values with source and training data
	#Exract the Forecast values and confidence intervals
	pd<-merge(ds,dfcastn,all.x=T) #final data.frame for use in ggplot


To leave a comment for the author, please follow the link and comment on their blog: Frank Davenport's Blog on R, Statistics, and all Things Spatial - R. offers daily e-mail updates about R news and tutorials on topics such as: Data science, Big Data, R jobs, visualization (ggplot2, Boxplots, maps, animation), programming (RStudio, Sweave, LaTeX, SQL, Eclipse, git, hadoop, Web Scraping) statistics (regression, PCA, time series, trading) and more...

If you got this far, why not subscribe for updates from the site? Choose your flavor: e-mail, twitter, RSS, or facebook...

Tags: ,

Comments are closed.

Search R-bloggers


Never miss an update!
Subscribe to R-bloggers to receive
e-mails with the latest R posts.
(You will not see this message again.)

Click here to close (This popup will not appear again)