我用hclust制作了一个dendogram,并将其切成两簇。我从图中知道哪一行对应于哪个集群。我想做的是在数据帧中创建一个单独的列,如果该行对应于第一个聚类,则该列将包含元素"class-1",如果对应于第二个聚类,则将包含元素"class-2"。
如果没有示例数据集,我将使用内置USArrests
。
如果创建带有标签的类factor
列"class-1"
"class-2"
R 将自动将它们分别分配给值1
和2
。
hc <- hclust(dist(USArrests), "ave") # taken from the help page ?hclust
memb <- cutree(hc, k = 2) #
res <- cbind(USArrests, Class = factor(unname(memb), labels = c("class-1", "class-2")))
head(res)
# Murder Assault UrbanPop Rape Class
#Alabama 13.2 236 58 21.2 class-1
#Alaska 10.0 263 48 44.5 class-1
#Arizona 8.1 294 80 31.0 class-1
#Arkansas 8.8 190 50 19.5 class-2
#California 9.0 276 91 40.6 class-1
#Colorado 7.9 204 78 38.7 class-2