9

如何使用 R 将 RMSE、斜率、截距和 r^2 添加到绘图中？我附上了一个带有示例数据的脚本，它与我的真实数据集的格式相似——不幸的是，我处于静止状态。有没有比从方程式创建对象并将其插入到图表中更简单的方法将这些统计信息添加到图表中text()？理想情况下，我希望统计数据显示在图表上。我怎样才能做到这一点？

## Generate Sample Data
x = c(2,4,6,8,9,4,5,7,8,9,10)
y = c(4,7,6,5,8,9,5,6,7,9,10)

# Create a dataframe to resemble existing data
mydata = data.frame(x,y)

#Plot the data
plot(mydata$x,mydata$y)
abline(fit <- lm(y~x))

# Calculate RMSE
model = sqrt(deviance(fit)/df.residual(fit))

# Add RMSE value to plot
text(3,9,model)
4

1 回答 1

17

这是一个使用基本图形并?plotmath绘制绘图并对其进行注释的版本

## Generate Sample Data
x = c(2,4,6,8,9,4,5,7,8,9,10)
y = c(4,7,6,5,8,9,5,6,7,9,10)

## Create a dataframe to resemble existing data
mydata = data.frame(x,y)

## fit model
fit <- lm(y~x, data = mydata)

接下来计算要出现在注释中的值。我更喜欢bquote()这一点，其中任何标记的内容.(foo)都将替换为 object 的值foo。答案@mnel 在评论中指出您使用substitute()不同的方式实现相同的目标。因此，我在工作区中为您可能希望在注释中显示的每个值创建对象：

## Calculate RMSE and other values
rmse <- round(sqrt(mean(resid(fit)^2)), 2)
coefs <- coef(fit)
b0 <- round(coefs[1], 2)
b1 <- round(coefs[2],2)
r2 <- round(summary(fit)$r.squared, 2)

现在使用描述的构造建立方程?plotmath：

eqn <- bquote(italic(y) == .(b0) + .(b1)*italic(x) * "," ~~ 
                  r^2 == .(r2) * "," ~~ RMSE == .(rmse))

完成后，您可以绘制绘图并用您的表达式对其进行注释

## Plot the data
plot(y ~ x, data = mydata)
abline(fit)
text(2, 10, eqn, pos = 4)

这使：

在此处输入图像描述

于 2012-10-29T10:06:36.520 回答