我绘制了以下数据并添加了黄土平滑器。我想在图中添加一个三阶多项式及其方程(包括残差)。有什么建议吗?
set.seed(1410)
dsmall<-diamonds[sample(nrow(diamonds), 100), ]
df<-data.frame("x"=dsmall$carat, "y"=dsmall$price)
p <-ggplot(df, aes(x, y))
p <- p + geom_point(alpha=2/10, shape=21, fill="blue", colour="black", size=5)
#Add a loess smoother
p<- p + geom_smooth(method="loess",se=TRUE)
如何添加三阶多项式?我尝试过:
p<- p + geom_smooth(method="lm", se=TRUE, fill=NA,formula=lm(y ~ poly(x, 3, raw=TRUE)),colour="red")
最后如何将三阶多项式方程和残差添加到图中? 我尝试过:
lm_eqn = function(df){
m=lm(y ~ poly(x, 3, df))#3rd degree polynomial
eq <- substitute(italic(y) == a + b %.% italic(x)*","~~italic(r)^2~"="~r2,
list(a = format(coef(m)[1], digits = 2),
b = format(coef(m)[2], digits = 2),
r2 = format(summary(m)$r.squared, digits = 3)))
as.character(as.expression(eq))
}
data.label <- data.frame(x = 1.5,y = 10000,label = c(lm_eqn(df)))
p<- p + geom_text(data=data.label,aes(x = x, y = y,label =label), size=8,family="Times",face="italic",parse = TRUE)
最佳答案
第 1 部分:要拟合多项式,请使用参数:
method=lm
- 您的操作正确formula=y ~ poly(x, 3, raw=TRUE)
- 即不要将其包含在对lm
的调用中
代码:
p + stat_smooth(method="lm", se=TRUE, fill=NA,
formula=y ~ poly(x, 3, raw=TRUE),colour="red")
<小时/>
第 2 部分:添加方程:
- 修改您的函数
lm_eqn()
以正确指定lm
的数据源 - 您的右括号位于错误的位置 - 使用
annotate()
来定位标签,而不是geom_text
代码:
lm_eqn = function(df){
m=lm(y ~ poly(x, 3), df)#3rd degree polynomial
eq <- substitute(italic(y) == a + b %.% italic(x)*","~~italic(r)^2~"="~r2,
list(a = format(coef(m)[1], digits = 2),
b = format(coef(m)[2], digits = 2),
r2 = format(summary(m)$r.squared, digits = 3)))
as.character(as.expression(eq))
}
p + annotate("text", x=0.5, y=15000, label=lm_eqn(df), hjust=0, size=8,
family="Times", face="italic", parse=TRUE)
关于r - 将三阶多项式及其方程添加到 r 中的 ggplot 中,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/11949331/