减少树的节点数,以获取具有多个子节点的节点

时间:2018-06-06 16:59:31

标签: r tree

以下树:

enter image description here

已从以下矩阵

获得
> mat
7  23 47 41 31
7  23 53 41 31
7  23 53 41 37
7  29 47 41 31
7  29 47 41 37
7  29 53 41 31
7  29 53 41 37
11 29 53 41 31
11 29 53 41 37

将“mat”的每一列作为树的一个级别。如果'data'是存储矩阵'mat'的数据帧

V1 V2 V3 V4 V5
7  23 47 41 31
7  23 53 41 31
7  23 53 41 37
7  29 47 41 31
7  29 47 41 37
7  29 53 41 31
7  29 53 41 37
11 29 53 41 31
11 29 53 41 37

生成上面树的代码如下

> data$pathString<-paste("0", data$V1,data$V2,data$V3,data$V4,data$V5,sep = "/")
> p_tree <- as.Node(data)
> export_graph(ToDiagrammeRGraph(p_tree), "tree.png")

我想按如下方式修改树:(1)如果级别为'n'的节点(由数字x标记)只有一个子节点在级别'n + 1',标记为数字y,那么程序将这两个节点汇集在一个节点中,该节点由产品x * y的结果标记; 2)如果级别为“n + 1”的节点没有子节点,则程序不执行任何操作并从另一个分支重新开始; 3)如果级别为“n + 1”的节点具有多个子节点,则程序应用点(1)并从每个子节点重新开始。

例如,对于我们示例的树,代码应该:

  • 将标有31 * 41 * 47 = 59737
  • 的节点替换为红色圆圈的节点
  • 将标有53 * 41 = 2173
  • 的节点替换为橙色圆圈的节点
  • 将标有47 * 41 = 1927
  • 的节点替换为绿色圆圈的节点
  • 将标有11 * 29 * 53 * 41 = 693187的节点替换为蓝色圆圈的节点

enter image description here

1 个答案:

答案 0 :(得分:7)

试试这个:

  freq <- sapply(1:ncol(data), function(x) {
  df <- data[, 1:x, drop = FALSE]

  cc <- aggregate(df[, 1], as.list(df), FUN = length)
  merge(df, cc, by = colnames(df), sort = FALSE)[, "x"]
  })

data$pathString <- sapply(1:nrow(data), function(x) {
  g <- 1
  for(i in 2:ncol(freq)) g <- c(g, 
        if(freq[x, i] == freq[x, i - 1]) g[i - 1] else g[i - 1] + 1)

  paste0(c("0", tapply(unlist(data[x, , drop = TRUE]), g, prod)), collapse = "/")
})


p_tree <- as.Node(data)

plot(p_tree)

enter image description here