使用R语言igraph包在树状图结构中向下传播节点信息

我来帮你搞定这个问题！你的需求是在没有层级信息的树状图里，把上层节点的info向下传递给下层缺失该信息的节点对吧？其实用igraph的遍历功能就能轻松实现，核心思路是从树的根节点出发，按顺序遍历所有节点，沿途跟踪当前有效的info值——遇到有自己info的节点就更新这个值，没有的话就继承当前的有效信息，这样就能自动完成向下传播了。

下面是具体的实现步骤和代码：

1. 先确认图的结构并找到根节点 #

首先我们需要找到树的根（也就是入度为0的节点，因为你的图是有向的父→子结构）：

library(data.table)
library(igraph)

# 你的原始数据
vertex <- data.table(ID = c("main","1_1","1_2","2_1","2_2","2_3","2_4","3_1","3_2","3_3","3_4"),
                     info = c(NA,"plouf","plof",NA,NA,"truc",NA,NA,NA,NA,NA))

edges <- data.table(to = c("main","main","1_1","1_1","1_2","1_2","2_1","2_1","2_3","2_3"),
                    from = c("1_1","1_2","2_1","2_2","2_3","2_4","3_1","3_2","3_3","3_4"))

g <- graph_from_data_frame(d = edges, vertices = vertex, directed = T)

# 找到根节点（入度为0的节点）
root <- V(g)[degree(g, mode = "in") == 0]$name

2. 按遍历顺序处理节点信息 #

我们用广度优先遍历（BFS）来按层级顺序访问节点（你也可以用深度优先DFS，结果是一样的），然后遍历每个节点并传递信息：

# 获取从根开始的广度优先遍历顺序
traverse_order <- bfs(g, root, order = "out", unreachable = FALSE)$order
traverse_nodes <- V(g)[traverse_order]$name

# 初始化最终info向量，用原始数据填充
final_info <- vertex$info
names(final_info) <- vertex$ID

# 跟踪当前有效的info值，初始为根节点的info（这里是NA）
current_info <- final_info[root]

# 遍历每个节点，传递信息
for (node in traverse_nodes) {
  if (!is.na(final_info[node])) {
    # 如果当前节点有自己的info，更新当前有效信息
    current_info <- final_info[node]
  } else {
    # 否则继承当前的有效信息
    final_info[node] <- current_info
  }
}

3. 生成最终结果表 #

把处理后的final_info合并回原始的节点表：

vertex_final <- vertex[, .(ID, info = final_info[ID])]

# 查看结果
print(vertex_final)

运行后你会得到和预期完全一致的结果：

ID  info
 1: main  <NA>
 2:  1_1 plouf
 3:  1_2  plof
 4:  2_1 plouf
 5:  2_2 plouf
 6:  2_3  truc
 7:  2_4  plof
 8:  3_1 plouf
 9:  3_2 plouf
10:  3_3  truc
11:  3_4  truc

这个方法完全不需要依赖层级信息，只靠图的父子关系就能自动完成信息的向下传播，不管你的树有多少层都能适用~

备注：内容来源于stack exchange，提问作者denis