如何让geom_smooth()趋势线穿过所有数据点并设置外部约束？

一、让趋势线精确穿过所有数据点 #

对于小数据集，若要让拟合线严格穿过每个数据点，你有两种常用方案：

1. 调整loess的span参数 #

geom_smooth(method = "loess")的span参数控制局部拟合的邻域比例，值越小，拟合线越贴近原始数据。对于小数据集（比如n<20），设置一个极小的span值（如0.1）即可让线穿过所有点：

library(ggplot2)
set.seed(123)
df <- data.frame(x = 1:10, y = rnorm(10, 1:10, 0.5))

ggplot(df, aes(x, y)) +
  geom_point(size = 2) +
  geom_smooth(method = "loess", span = 0.1, se = FALSE, color = "blue")

2. 使用样条插值（更可靠） #

如果需要确保绝对穿过所有点，样条插值是更直接的选择。你可以用spline()函数生成密集的插值点，再用geom_line()绘制：

# 生成样条插值点
spline_df <- as.data.frame(spline(df$x, df$y, n = 100))

ggplot(df, aes(x, y)) +
  geom_point(size = 2) +
  geom_line(data = spline_df, aes(x, y), color = "red", linewidth = 1)

也可以结合gam方法，设置样条自由度等于数据点数量，强制曲线穿过所有点：

ggplot(df, aes(x, y)) +
  geom_point() +
  stat_smooth(method = "gam", formula = y ~ s(x, k = nrow(df)), se = FALSE, color = "green")

二、为趋势线添加外部约束 #

如果需要限制趋势线不超过某个特定值（或其他约束），可以通过以下方式实现：

1. 手动裁剪拟合结果 #

先完成基础拟合，再对预测值应用约束（比如限制最大值为10）：

# 拟合loess模型
loess_mod <- loess(y ~ x, data = df, span = 0.1)
# 生成预测序列
pred_df <- data.frame(x = seq(min(df$x), max(df$x), length.out = 100))
pred_df$y_pred <- predict(loess_mod, newdata = pred_df)
# 应用约束：不超过10
pred_df$y_constrained <- pmin(pred_df$y_pred, 10)

ggplot(df, aes(x, y)) +
  geom_point(size = 2) +
  geom_line(data = pred_df, aes(x, y_constrained), color = "purple", linewidth = 1) +
  geom_hline(yintercept = 10, linetype = "dashed", color = "gray")

2. 使用带约束的回归模型（严谨方案） #

如果需要更严谨的约束（比如拟合过程中就遵守限制，而非事后裁剪），可以用mgcv包的pcls函数设置不等式约束：

library(mgcv)

# 定义约束：所有拟合值 <=10
x_seq <- seq(min(df$x), max(df$x), length.out = 100)
# 生成样条基函数
b <- s(x_seq, k = nrow(df), bs = "cr")$X
# 构建约束矩阵：-b %*% coef >= -10（等价于b %*% coef <=10）
constraints <- list(
  A = -b,
  b = rep(-10, nrow(b))
)

# 拟合带约束的样条
fit_pcls <- pcls(b, df$y, constraints$A, constraints$b, method = "BFGS")
pred_df$y_pcls_constrained <- as.vector(b %*% fit_pcls)

ggplot(df, aes(x, y)) +
  geom_point(size = 2) +
  geom_line(data = pred_df, aes(x, y_pcls_constrained), color = "orange", linewidth = 1) +
  geom_hline(yintercept = 10, linetype = "dashed", color = "gray")

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