Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Plotting functions on top of datapoints in R

Is there a way of overlaying a mathematical function on top of data using ggplot?

## add ggplot2
library(ggplot2)

# function
eq = function(x){x*x}

# Data                     
x = (1:50)     
y = eq(x)                                                               

# Make plot object    
p = qplot(    
x, y,   
xlab = "X-axis", 
ylab = "Y-axis",
) 

# Plot Equation     
c = curve(eq)  

# Combine data and function
p + c #?

In this case my data is generated using the function, but I want to understand how to use curve() with ggplot.

like image 372
womble Avatar asked Dec 05 '09 22:12

womble


2 Answers

You probably want stat_function:

library("ggplot2")
eq <- function(x) {x*x}
tmp <- data.frame(x=1:50, y=eq(1:50))

# Make plot object
p <- qplot(x, y, data=tmp, xlab="X-axis", ylab="Y-axis")
c <- stat_function(fun=eq)
print(p + c)

and if you really want to use curve(), i.e., the computed x and y coordinates:

qplot(x, y, data=as.data.frame(curve(eq)), geom="line")
like image 183
rcs Avatar answered Oct 25 '22 06:10

rcs


Given that your question title is "plotting functions in R", here's how to use curve to add a function to a base R plot.

Create data as before

eq = function(x){x*x}; x = (1:50); y = eq(x)

Then use plot from base graphics to plot the points followed by curve with the add=TRUE argument, to add the curve.

plot(x, y,  xlab = "X-axis", ylab = "Y-axis") 
curve(eq, add=TRUE)
like image 42
Jeromy Anglim Avatar answered Oct 25 '22 08:10

Jeromy Anglim