在R ggplot中为等高线添加函数增长方向指示箭头

要给等高线添加指示函数增长方向（梯度方向）的箭头，核心是先获取等高线的坐标点，再计算每个点的梯度向量，最后用线段加箭头的方式绘制。以下是基于你提供代码的完整实现：

步骤说明 #

1. 提取等高线坐标：通过ggplot_build获取stat_contour生成的等高线点数据；
2. 计算梯度向量：根据函数的偏导数计算每个点的梯度（梯度方向即函数增长最快的方向）；
3. 归一化梯度向量：统一箭头长度，避免因梯度大小差异导致箭头长短不一；
4. 绘制箭头：用geom_segment结合arrow()函数添加箭头。

完整代码 #

library(tidyverse)

# 原始数据与函数定义
x <- seq(1,2,length.out=100)
y <- seq(1,2,length.out=100)
myf <- function(x,y) {x*y}
myg <- function(x,y) {x^2 + y^2}

d1 <- expand.grid(X1 = x, X2 = y) %>%
  mutate(Z = myf(X1,X2)) %>%
  as.data.frame()
d2 <- expand.grid(X1 = x, X2 = y) %>%
  mutate(Z = myg(X1,X2)) %>%
  as.data.frame()

# 绘制基础等高线图
p <- ggplot(data = d1, aes(x=X1,y=X2,z=Z))+
  stat_contour(breaks = c(2)) +
  stat_contour(data=d2, aes(x=X1,y=X2,z=Z), breaks=c(6))

# 提取两条等高线的坐标数据
contour_f <- ggplot_build(p)$data[[1]]  # myf函数的等高线（Z=2）
contour_g <- ggplot_build(p)$data[[2]]  # myg函数的等高线（Z=6）

# 处理myf函数的梯度箭头：梯度为(y, x)
contour_f <- contour_f %>%
  mutate(
    grad_x = y,
    grad_y = x,
    mag = sqrt(grad_x^2 + grad_y^2),  # 计算梯度向量的模长
    end_x = x + grad_x/mag * 0.05,    # 箭头终点x（0.05为箭头长度，可调整）
    end_y = y + grad_y/mag * 0.05     # 箭头终点y
  )

# 处理myg函数的梯度箭头：梯度为(2x, 2y)
contour_g <- contour_g %>%
  mutate(
    grad_x = 2*x,
    grad_y = 2*y,
    mag = sqrt(grad_x^2 + grad_y^2),
    end_x = x + grad_x/mag * 0.05,
    end_y = y + grad_y/mag * 0.05
  )

# 添加箭头并显示最终图形
p +
  geom_segment(data = contour_f,
               aes(x = x, y = y, xend = end_x, yend = end_y),
               arrow = arrow(length = unit(0.1, "cm")),
               color = "blue") +
  geom_segment(data = contour_g,
               aes(x = x, y = y, xend = end_x, yend = end_y),
               arrow = arrow(length = unit(0.1, "cm")),
               color = "red")

细节调整 #

• 箭头长度：修改0.05可以调整箭头的延伸长度，数值越大箭头越长；
• 箭头样式：arrow()函数可添加type="open"等参数调整箭头样式；
• 箭头密度：如果觉得箭头太密集，可以对contour_f和contour_g进行抽样，比如用slice_sample(n=20)保留部分点绘制箭头。

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