R画树状图：一种轻量级方法

树状图是用于表示层次关系的图，例如从分层聚类获得的层次关系。它们常用于生物学，尤其是遗传学，用于揭示一组基因或者分群的关系。

R中已有一些方法可以画树形图（请查看这里），比如基础R或ape包。 对于基于ggplot2的解决方案，让我们提一下ggdendro，dendextend或ggtree。
ggdendro是稳定的，轻量级的（除了MASS和ggplot2之外没有依赖性），并且允许以方便的格式访问聚类数据，但其功能在可视化方面有点局限。 另一方面，dendextend和ggtree提供了许多强大的功能，但代价是更高的依赖性要求和更陡峭的学习曲线才能有效地使用它们。

我想要一个“轻量级”且灵活的基于ggplot2的解决方案来绘制树形图，特别是可以使用不同的分支颜色突出显示聚类。
受到这个stackoverflow问题的启发，我终于完成了使用ggdendro和ggplot2编写下面描述的函数。

1library(ggdendro)
2library(ggplot2)

调整ggdendro

首先，我’扩展’了ggdendro :: dendro_data（）。 dendro_data_k（）函数接受一个k参数，一个整数，指定所需簇的数量。 此值仅用于base :: cutree（）函数，并且对于每个集群，将根据其x，xend和yend坐标为这些段分配相应叶子的集群ID。 这可能不是最优雅的方式，但它非常简单。

 1dendro_data_k <- function(hc, k) {
 2
 3  hcdata    <-  ggdendro::dendro_data(hc, type = "rectangle")
 4  seg       <-  hcdata$segments
 5  labclust  <-  cutree(hc, k)[hc$order]
 6  segclust  <-  rep(0L, nrow(seg))
 7  heights   <-  sort(hc$height, decreasing = TRUE)
 8  height    <-  mean(c(heights[k], heights[k - 1L]), na.rm = TRUE)
 9
10  for (i in 1:k) {
11    xi      <-  hcdata$labels$x[labclust == i]
12    idx1    <-  seg$x    >= min(xi) & seg$x    <= max(xi)
13    idx2    <-  seg$xend >= min(xi) & seg$xend <= max(xi)
14    idx3    <-  seg$yend < height
15    idx     <-  idx1 & idx2 & idx3
16    segclust[idx] <- i
17  }
18
19  idx                    <-  which(segclust == 0L)
20  segclust[idx]          <-  segclust[idx + 1L]
21  hcdata$segments$clust  <-  segclust
22  hcdata$segments$line   <-  as.integer(segclust < 1L)
23  hcdata$labels$clust    <-  labclust
24
25  hcdata
26}

绘图函数

通过从ggdendro和上面的函数获得的所需的数据结构，可以使用ggplot2构建树。 使用两个geom：分支的geom_segment（）和标签的geom_text（）。
如果我们想要自定义树的方向（例如从上到下或从左到右）或格式（圆形图），事情会变得有点复杂。 为了更容易地处理它，使用（内部）不同的函数来设置标签的参数（角度，偏移，……）。

 1set_labels_params <- function(nbLabels,
 2                              direction = c("tb", "bt", "lr", "rl"),
 3                              fan       = FALSE) {
 4  if (fan) {
 5    angle       <-  360 / nbLabels * 1:nbLabels + 90
 6    idx         <-  angle >= 90 & angle <= 270
 7    angle[idx]  <-  angle[idx] + 180
 8    hjust       <-  rep(0, nbLabels)
 9    hjust[idx]  <-  1
10  } else {
11    angle       <-  rep(0, nbLabels)
12    hjust       <-  0
13    if (direction %in% c("tb", "bt")) { angle <- angle + 45 }
14    if (direction %in% c("tb", "rl")) { hjust <- 1 }
15  }
16  list(angle = angle, hjust = hjust, vjust = 0.5)
17}

 1plot_ggdendro <- function(hcdata,
 2                          direction   = c("lr", "rl", "tb", "bt"),
 3                          fan         = FALSE,
 4                          scale.color = NULL,
 5                          branch.size = 1,
 6                          label.size  = 3,
 7                          nudge.label = 0.01,
 8                          expand.y    = 0.1) {
 9
10  direction <- match.arg(direction) # if fan = FALSE
11  ybreaks   <- pretty(segment(hcdata)$y, n = 5)
12  ymax      <- max(segment(hcdata)$y)
13
14  ## branches
15  p <- ggplot() +
16    geom_segment(data         =  segment(hcdata),
17                 aes(x        =  x,
18                     y        =  y,
19                     xend     =  xend,
20                     yend     =  yend,
21                     linetype =  factor(line),
22                     colour   =  factor(clust)),
23                 lineend      =  "round",
24                 show.legend  =  FALSE,
25                 size         =  branch.size)
26
27  ## orientation
28  if (fan) {
29    p <- p +
30      coord_polar(direction = -1) +
31      scale_x_continuous(breaks = NULL,
32                         limits = c(0, nrow(label(hcdata)))) +
33      scale_y_reverse(breaks = ybreaks)
34  } else {
35    p <- p + scale_x_continuous(breaks = NULL)
36    if (direction %in% c("rl", "lr")) {
37      p <- p + coord_flip()
38    }
39    if (direction %in% c("bt", "lr")) {
40      p <- p + scale_y_reverse(breaks = ybreaks)
41    } else {
42      p <- p + scale_y_continuous(breaks = ybreaks)
43      nudge.label <- -(nudge.label)
44    }
45  }
46
47  # labels
48  labelParams <- set_labels_params(nrow(hcdata$labels), direction, fan)
49  hcdata$labels$angle <- labelParams$angle
50
51  p <- p +
52    geom_text(data        =  label(hcdata),
53              aes(x       =  x,
54                  y       =  y,
55                  label   =  label,
56                  colour  =  factor(clust),
57                  angle   =  angle),
58              vjust       =  labelParams$vjust,
59              hjust       =  labelParams$hjust,
60              nudge_y     =  ymax * nudge.label,
61              size        =  label.size,
62              show.legend =  FALSE)
63
64  # colors and limits
65  if (!is.null(scale.color)) {
66    p <- p + scale_color_manual(values = scale.color)
67  }
68
69  ylim <- -round(ymax * expand.y, 1)
70  p    <- p + expand_limits(y = ylim)
71
72  p
73}
74

基本的树状图

我们现在准备建立一个树状图。 默认情况下，应用ggplot2的标准主题。

 1mtc <- scale(mtcars)
 2D   <- dist(mtc)
 3hc  <- hclust(D)
 4
 5hcdata <- dendro_data_k(hc, 3)
 6
 7p <- plot_ggdendro(hcdata,
 8                   direction   = "lr",
 9                   expand.y    = 0.2)
10p

定制的树状图

我们可以通过调整Plot_ggdendro（）参数或添加其他属性来进一步自定义树形图。 下面是ggplot2 :: theme_void（）的示例。