如何在R的ggdag包绘制的DAG箭头上方添加文本？

完全可以实现箭头上方添加文本的效果，以下是两种实用方案：

方法1：使用ggdag自带的边标签函数 #

ggdag提供了专门的工具给箭头加标签，你可以在构建DAG时直接指定每条边的标签，再用ggdag_edge_label绘制：

library(ggdag)
library(ggplot2)

# 构建DAG并指定每条边的标签
example <- dagify(
  d ~ p,
  m ~ d,
  y ~ d,
  y ~ m,
  edge_labels = c("p->d" = "暴露", "d->m" = "中介", "d->y" = "直接效应", "m->y" = "间接效应")
)

# 生成带箭头标签的DAG图
ggdag_edge_label(example) + theme_dag()

如果需要调整标签的位置、大小样式，可以拆分图层单独控制：

ggdag(example) +
  geom_dag_edge() +
  geom_dag_node() +
  geom_dag_text() +
  geom_dag_edge_label(aes(label = edge_label), vjust = -0.5, size = 4) + # 调整垂直偏移和字号
  theme_dag()

方法2：手动提取边数据添加文本 #

如果需要更灵活的布局控制，可以先提取DAG的坐标数据，再用ggplot2的geom_text手动添加标签：

library(ggdag)
library(ggplot2)

example <- dagify(d ~ p, m ~ d, y ~ d, y ~ m)

# 提取DAG的布局坐标
dag_layout <- ggdag_layout(example)
# 整理边的起点、终点坐标
edge_data <- dag_layout %>% 
  filter(!is.na(to)) %>% 
  left_join(dag_layout %>% select(name, x_end = x, y_end = y), by = c("to" = "name"))

# 绘制DAG并手动放置标签
ggdag(example) +
  geom_text(
    data = edge_data,
    aes(x = (x + x_end)/2, y = (y + y_end)/2 + 0.1, label = c("暴露", "中介", "直接效应", "间接效应")),
    size = 4
  ) +
  theme_dag()

这里通过计算边中点的上方位置（y = (y + y_end)/2 + 0.1）放置标签，你可以根据图的布局调整偏移量。

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