用 R 绘制决策树

Question

我有一个 440*2 的矩阵，如下所示：

1   144
1   152
1   135
2   3
2   12
2   107
2   31
3   4
3   147
3   0
4   end
4   0
4   0
5   6
5   7
5   10
5   9

左列是起点，例如在应用程序中，左侧的所有 1 都在同一页面上。它们导致三个选择，第 144,152,135 页。然后这些页面可以各自指向另一个页面，依此类推，直到右侧列显示“结束”。我想要的是一种可视化这棵树的规模的方法。我意识到考虑到 nb 行，它会非常大，所以可能对图形不友好，所以为了清楚起见，我想知道总共有多少条可能的路线（从每个起点，到它提供的每个选项以及每个选项的最终目的地. 我意识到会有重叠，但这就是为什么我觉得这很难计算）。

其次，每个数字都有一个关联的标题。我想要一个函数，如果你输入一个给定的标题，它将绘制所有可能的起点及其相关的路径。这应该小得多，因此图形友好。

例如

dta <- "
14  12  as
186  187  Frac
187  154  Low
23   52   Med
52   11   Lip
15  55  asd
11   42   AAA
42   154   BBB
154   end  Coll"

编辑示例数据以显示某些分支未连接到所需的树

  dta <- "
  14  12  as
  186  187  Frac
  187  154  Low
  23   52   Med
  52   11   Lip
  11   42   AAA
  42   154   BBB
  154   end  Coll"

 dta <- gsub("   ", ",", dta, fixed = TRUE)   
 dta <- gsub("  ", ",", dta, fixed = TRUE)

df <- read.csv(textConnection(dta), stringsAsFactors = FALSE, header = FALSE)
names(df) <- c("from", "to", "nme")
library(data.tree)
Warning message:
package ‘data.tree’ was built under R version 3.2.5
tree <- FromDataFrameNetwork(df)
 **Error in FromDataFrameNetwork(df) :**
  **Cannot find root name. network is not a tree!**

我做了这个例子来展示第 1 列如何导致第 2 列中的值，然后引用第 1 列中的值，直到你到达末尾。不同的起点最终会导致到达同一目的地的不同长度的路径。所以这看起来像：

所以在这里，我想看看你如何从所有起点走到“科尔”

非常感谢任何帮助

Answer 1

如果你确实有一棵树（例如没有循环），你可以使用 data.tree：

首先转换为data.frame：

 dta <- "
14  12  as
186  187  Frac
187  154  Low
23   52   Med
52   11   Lip
15  55  asd
11   42   AAA
42   154   BBB
154   end  Coll
55  end  efg
12  end  hij"

dta <- gsub("   ", ",", dta, fixed = TRUE)
dta <- gsub("  ", ",", dta, fixed = TRUE)


df <- read.csv(textConnection(dta), stringsAsFactors = FALSE, header = FALSE)
names(df) <- c("from", "to", "nme")

现在，转换为 data.tree：

library(data.tree)
tree <- FromDataFrameNetwork(df)

tree$leafCount

您现在可以导航到任何子树进行分析和绘图。例如。使用以下任何一种可能性：

subTree <- tree$FindNode(187)
subTree <- Climb(tree, nme = "Coll", nme = "Low")
subTree <- tree$`154`$`187`

subTree <- Clone(tree$`154`)

也许您只需要打印：

print(subTree , "nme")

这将像这样打印：

  levelName          nme
1 154                Coll
2  ¦--187             Low
3  ¦   °--186        Frac
4  °--42              BBB
5      °--11          AAA
6          °--52      Lip
7              °--23  Med

否则，请使用精美的绘图：

SetNodeStyle(subTree , style = "filled,rounded", shape = "box", fontname = "helvetica", label = function(node) node$nme, tooltip = "name")
plot(subTree , direction = "descend")

看起来像这样：