Как я могу изменить цвет на переменную, отличную от номера кластера в fviz_silhouette

Я использую пакет factoextra в R для создания графика силуэта. В настоящее время силуэт автоматически окрашивает ваш график с помощью кластеризации. Я хочу раскрасить другую переменную Site, которую я определил как x. Я пробовал и заливку, и цвет, изменяя его на переменную сайта, но ничего не работает. Я также пытался использовать scale_color_manual и scale_fill_discrete. Я думаю, что ключ находится в scale_fill_discrete, как указывает исходный код от другого пользователя.

mapping <- aes_string(x = "name", y = "sil_width", color = "cluster", fill = "cluster")

Мне в основном нужно изменить color="cluster" на colour= "x".
Я вернул окончательный код построения графика к его самой простой форме.

pamspec <- pam(spec, 3, keep.diss = TRUE) 
plot(pamspec)

spec <- cbind(pamspec$clustering)
autoplot(pam(spec,3), frame=TRUE, frame.type = "norm")

pamspec$site <- spec$Site
x <- pamspec$site
fviz_silhouette(pamspec, label=TRUE) + theme_classic()

r cluster-computing ggplot2 silhouette
person Sophie Williams    schedule 15.05.2020    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Я предлагаю следующую модифицированную версию fviz_silhouette с дополнительным входным аргументом var.col.

myfviz_silhouette <- function (sil.obj, var.col, label = FALSE, print.summary = TRUE, ...) {
    if (inherits(sil.obj, c("eclust", "hcut", "pam", "clara", 
        "fanny"))) {
        df <- as.data.frame(sil.obj$silinfo$widths, stringsAsFactors = TRUE)
    }
    else if (inherits(sil.obj, "silhouette")) 
        df <- as.data.frame(sil.obj[, 1:3], stringsAsFactors = TRUE)
    else stop("Don't support an oject of class ", class(sil.obj))
    df <- df[order(df$cluster, -df$sil_width), ]
    if (!is.null(rownames(df))) 
        df$name <- factor(rownames(df), levels = rownames(df))
    else df$name <- as.factor(1:nrow(df))
    df$cluster <- as.factor(df$cluster)
    df$var_col <- var.col
    mapping <- aes_string(x = "name", y = "sil_width", color = "var_col", 
        fill = "var_col")
    p <- ggplot(df, mapping) + geom_bar(stat = "identity") + 
        labs(y = "Silhouette width Si", x = "", title = paste0("Clusters silhouette plot ", 
            "\n Average silhouette width: ", round(mean(df$sil_width), 
                2))) + ggplot2::ylim(c(NA, 1)) + geom_hline(yintercept = mean(df$sil_width), 
        linetype = "dashed", color = "red")
    p <- ggpubr::ggpar(p, ...)
    if (!label) 
        p <- p + theme(axis.text.x = element_blank(), axis.ticks.x = element_blank())
    else if (label) 
        p <- p + theme(axis.text.x = element_text(angle = 45))
    ave <- tapply(df$sil_width, df$cluster, mean)
    n <- tapply(df$cluster, df$cluster, length)
    sil.sum <- data.frame(cluster = names(ave), size = n, ave.sil.width = round(ave, 
        2), stringsAsFactors = TRUE)
    if (print.summary) 
        print(sil.sum)
    p
}

Вот пример его использования:

library(factoextra)
library(cluster)
pamspec <- pam(iris[,-5], 3, keep.diss = TRUE) 

color_var <- iris$Species
myfviz_silhouette(pamspec, color_var, label=TRUE) +
   theme_classic()

введите здесь описание изображения

person Marco Sandri    schedule 01.06.2020