R语言Plotly Sankey图问题：消除循环节点且保留完整数据

问题根源 #

Plotly自动处理空值时会生成空白节点，这些节点易被错误关联导致循环；直接删除空白节点会误删关联的有效链路（比如物种2的后续数据），核心是要提前控制空值的处理逻辑，手动管理节点与链接的映射关系。

具体解决步骤 #

1. 预处理数据：主动标记空值为终止节点

不要让Plotly自动处理NA，把空值替换为明确的"终止"标识（比如"End"），同时过滤掉从终止节点出发的无效链路：

# 假设原始数据框为sankey_data，包含source、target、value列
sankey_data <- sankey_data %>%
  mutate(
    target = ifelse(is.na(target), "End", target),
    source = ifelse(source == "End", NA, source)
  ) %>%
  drop_na(source) # 删除源节点为NA的无效行

2. 手动构建节点与链接映射，避免自动生成空白节点

手动提取所有有效节点（排除"End"），创建节点名称到索引的映射，再构建符合Plotly要求的链接数据：

# 提取所有唯一有效节点（排除终止标记）
all_nodes <- unique(c(sankey_data$source, sankey_data$target))
all_nodes <- all_nodes[all_nodes != "End"]

# 创建节点名称到索引的映射（Plotly索引从0开始）
node_map <- setNames(seq_along(all_nodes) - 1, all_nodes)

# 转换链接数据为Plotly格式：将"End"设为单独的终点节点
plotly_links <- sankey_data %>%
  mutate(
    source = node_map[source],
    target = ifelse(target == "End", length(all_nodes), node_map[target])
  )

# 更新节点列表，加入"End"作为终点节点
all_nodes <- c(all_nodes, "End")

3. 精准控制节点配色，保留原有逻辑

手动指定节点颜色向量，第一列的目标节点（比如物种1、物种2）用设定颜色，其余节点（包括"End"）设为黑色或透明：

# 假设第一列节点为c("物种1", "物种2")，对应颜色c("red", "orange")
node_colors <- ifelse(all_nodes %in% c("物种1", "物种2"), 
                      case_when(
                        all_nodes == "物种1" ~ "red",
                        all_nodes == "物种2" ~ "orange"
                      ), 
                      "black")
# 隐藏终止节点：设为透明+空标签
node_colors[all_nodes == "End"] <- "rgba(0,0,0,0)"
node_labels <- ifelse(all_nodes == "End", "", all_nodes)

4. 绘制修正后的Sankey图

用手动构建的节点和链接数据绘图，确保链路正确终止且物种2的后续数据保留：

library(plotly)

plot_ly(
  type = "sankey",
  orientation = "h",
  node = list(
    label = node_labels,
    color = node_colors,
    pad = 15,
    thickness = 20
  ),
  link = list(
    source = plotly_links$source,
    target = plotly_links$target,
    value = plotly_links$value,
    color = "rgba(0,0,0,0.2)" # 链路透明度可自定义
  )
) %>%
  layout(title = "修正后的Sankey图", font = list(size = 10))

关键说明 #

• 提前将空值标记为"End"，从根源上避免Plotly自动生成空白节点，解决循环问题
• 手动管理节点映射，确保"End"仅作为链路终点，不参与后续关联，同时完整保留物种2的有效链路
• 终止节点设为透明隐藏，不影响可视化效果，但能保证链路逻辑正确

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