如何在R的DiagrammeR图中分组节点并强制指定节点位置？

1. 实现节点分组与包围框 #

DiagrammeR基于Graphviz，要实现节点分组带包围框，需使用**子图（subgraph）**功能，而非单纯设置group属性。具体步骤：

• 先读取节点数据，按income字段分组
• 用add_subgraph()为每个分组创建子图，指定子图包含的节点ID，并设置子图的样式（如标签、边框）

2. 指定首行节点顺序 #

tree布局本质是Graphviz的dot布局，要控制首行节点顺序，可通过设置节点的rank属性，将首行节点设为同一层级，并调整显示顺序：

• 给首行节点添加rank = same属性，确保它们处于同一行
• 若自动布局顺序不符合预期，可添加无箭头的隐形边（style = invis）强制指定节点排列顺序

完整代码示例 #

library(DiagrammeR)
library(tidyverse)

# 先读取节点和边数据，避免直接从文件添加时丢失属性
nodes_df <- read_csv("data/tbm_nodes.csv")
edges_df <- read_csv("data/tbm_edges.csv")

# 初始化图对象
test <- create_graph() %>%
  # 添加节点并保留income属性
  add_nodes_from_table(
    table = nodes_df,
    label_col = label,
    extra_attrs = c("income")
  ) %>%
  add_edges_from_table(
    table = edges_df,
    from_col = from,
    to_col = to,
    from_to_map = id_external
  ) %>%
  set_node_attrs(shape, "box") %>%
  set_node_attrs(fixedsize, "false")

# 按income字段添加分组子图
for (group_val in unique(nodes_df$income)) {
  group_nodes <- nodes_df %>% filter(income == group_val) %>% pull(id)
  test <- test %>%
    add_subgraph(
      nodes = group_nodes,
      label = str_c("Income: ", group_val),
      style = "rounded,filled",
      color = "#888888",
      fillcolor = "#f0f0f0"
    )
}

# 指定首行节点顺序（替换为你的首行节点ID）
first_row_nodes <- c(1, 2, 3)
test <- test %>%
  # 将首行节点设为同一层级
  set_node_attrs_ws(
    nodes = first_row_nodes,
    attr = rank,
    value = "same"
  ) %>%
  # 添加隐形边强制顺序（自动布局顺序不符时启用）
  add_edge(from = 1, to = 2, style = "invis") %>%
  add_edge(from = 2, to = 3, style = "invis")

# 渲染流程图，可调整布局参数优化效果
render_graph(test, layout = "tree", rankdir = "TB", ranksep = 0.8, nodesep = 0.5)

关键说明 #

• 无需放弃tree布局，通过Graphviz的子图和rank属性即可满足需求
• group属性在Graphviz中主要用于边的聚合，而非节点分组包围框，因此之前的尝试无效
• set_node_position()在自动布局（如tree）下会被覆盖，用rank属性控制层级和顺序更可靠

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