r - 根据 R 中数据框的列数为树状图的末端分支(或叶子)着色

Question

来自数据帧 data.main ，我能够生成一个 hclust树状图，

aa1<- c(2,4,6,8)
bb1<- c(1,3,7,11)
aa2<-c(3,6,9,12)
bb2<-c(3,5,7,9)
data.main<- data.frame(aa1,bb1,aa2,bb2)
d1<-dist(t(data.main))
hcl1<- hclust(d1)
plot(hcl1)

此外，我知道有一些方法可以使用树的截断来为 Twig 或树叶着色。但是，是否可以根据部分列名或列号为它们着色(例如，我希望对应于 aa1 、 aa2 的分支为红色，而 bb1 和 bb2 为蓝色)？

我查过R包 dendextend但我仍然无法找到一种直接/简单的方法来获得所需的结果。

Answer 1

更改树状图的颜色比更改 hclust 对象更容易，但转换起来非常简单。你可以做

drg1 <- dendrapply(as.dendrogram(hcl1, hang=.1), function(n){
  if(is.leaf(n)){
    labelCol <- c(a="red", b="blue")[substr(attr(n,"label"),1,1)];
    attr(n, "nodePar") <- list(pch = NA, lab.col = labelCol);
    attr(n, "edgePar") <- list(col = labelCol); # to color branch as well
  }
  n;
});
plot(drg1)

这将绘制