如何为ggplot2指定分面设置匹配的stat_poly_eq回归方程与R²？

你遇到的核心问题是：stat_poly_eq默认使用固定公式计算回归方程和R²，完全不知道你在geom_smooth里为第6个分面自定义了拟合模型，所以自然会显示和实际拟合不匹配的结果。

要解决这个问题，我们可以给stat_poly_eq也加上和自定义smooth函数一样的「分面判断逻辑」，让它根据当前分面的编号动态选择对应的公式，实现拟合模型和标注内容的完全匹配。

完整解决方案代码 #

# 加载所需库
library(ggplot2)
library(ggpmisc)

# 自定义拟合函数：根据分面切换模型
custom.smooth <- function(formula, data,...){
 smooth.call <- match.call()
 panel_num <- as.numeric(unique(data$PANEL))
 if(panel_num == 6) {
  # 第6个分面使用y ~ log(x)的线性回归
  smooth.call[[1]] <- quote(lm)
  smooth.call$formula <- as.formula("y ~ log(x)")
 }else{
  # 其他分面使用默认y ~ x的线性回归
  smooth.call[[1]] <- quote(lm)
 }
 eval.parent(smooth.call)
}

# 自定义公式函数：为stat_poly_eq提供分面匹配的公式
custom_eq_formula <- function(data) {
  panel_num <- as.numeric(unique(data$PANEL))
  if (panel_num == 6) {
    y ~ log(x)
  } else {
    y ~ x
  }
}

# 绘制图形+匹配的回归标注
p <- ggplot(mtcars,aes(x = disp, y = mpg)) + 
  geom_point() + 
  facet_grid(gear ~ am) +
  geom_smooth(method = "custom.smooth", se = FALSE)

# 添加与拟合模型完全匹配的回归方程和R²
p + stat_poly_eq(
  formula = custom_eq_formula,
  aes(label = paste(..eq.label.., ..rr.label.., sep = "~~~")),
  parse = TRUE,
  label.x.npc = "right"
)

关键逻辑说明 #

1. 共享分面判断逻辑：不管是自定义拟合函数还是公式函数，都通过data$PANEL获取当前分面的编号（第6个分面对应的值为6），以此作为切换规则的依据，保证拟合和标注的逻辑完全一致。
2. 动态公式传递：stat_poly_eq的formula参数支持接收一个函数，这个函数会自动接收当前分面的数据集作为参数，我们利用这一点实现了分面级的公式切换。
3. 扩展性强：如果后续需要给更多分面设置不同的模型/公式，只需要扩展if-else分支即可（比如if(panel_num == 3) {y ~ x + I(x^2)}）。

内容的提问来源于stack exchange，提问作者Tianjian Qin