Как создать сюжет Marimekko/Mosaic в ggplot2

Question

График Marimekko/Mosaic является хорошим графиком по умолчанию, когда x и y являются категориальными переменными. Каков наилучший способ создать их с помощью ggplot?

Единственная ссылка, которую я смог найти, была в этом 4-летнем сообщении в блоге, но это кажется немного устаревшим Есть ли лучшие или более легкие реализации, доступные сейчас? Пакет GGally имеет функцию ggally_ratio но это производит что-то совсем другое:

Answer 1

Я сделал это сам некоторое время назад, используя только geom_barЯ превратил его в общую функцию, поэтому он должен работать на любых двух factors,

ggMMplot <- function(var1, var2){
  require(ggplot2)
  levVar1 <- length(levels(var1))
  levVar2 <- length(levels(var2))

  jointTable <- prop.table(table(var1, var2))
  plotData <- as.data.frame(jointTable)
  plotData$marginVar1 <- prop.table(table(var1))
  plotData$var2Height <- plotData$Freq / plotData$marginVar1
  plotData$var1Center <- c(0, cumsum(plotData$marginVar1)[1:levVar1 -1]) +
    plotData$marginVar1 / 2

  ggplot(plotData, aes(var1Center, var2Height)) +
    geom_bar(stat = "identity", aes(width = marginVar1, fill = var2), col = "Black") +
    geom_text(aes(label = as.character(var1), x = var1Center, y = 1.05)) 
  }

ggMMplot(diamonds$cut, diamonds$clarity)

Answer 2

У меня была такая же проблема для проекта некоторое время назад. Мое решение было использовать geom_bar вместе с scales="free_x", space="free_x" вариант в facet_grid чтобы приспособить бар различной ширины:

# using diamonds dataset for illustration
df <- diamonds %>%
  group_by(cut, clarity) %>%
  summarise(count = n()) %>%
  mutate(cut.count = sum(count),
         prop = count/sum(count)) %>%
  ungroup()

ggplot(df,
       aes(x = cut, y = prop, width = cut.count, fill = clarity)) +
  geom_bar(stat = "identity", position = "fill", colour = "black") +
  # geom_text(aes(label = scales::percent(prop)), position = position_stack(vjust = 0.5)) + # if labels are desired
  facet_grid(~cut, scales = "free_x", space = "free_x") +
  scale_fill_brewer(palette = "RdYlGn") +
  # theme(panel.spacing.x = unit(0, "npc")) + # if no spacing preferred between bars
  theme_void() 

Answer 3

Первая попытка Я не уверен, как поставить метки факторов на оси, хотя.

makeplot_mosaic <- function(data, x, y, ...){
  xvar <- deparse(substitute(x))
  yvar <- deparse(substitute(y))
  mydata <- data[c(xvar, yvar)];
  mytable <- table(mydata);
  widths <- c(0, cumsum(apply(mytable, 1, sum)));
  heights <- apply(mytable, 1, function(x){c(0, cumsum(x/sum(x)))});

  alldata <- data.frame();
  allnames <- data.frame();
  for(i in 1:nrow(mytable)){
    for(j in 1:ncol(mytable)){
      alldata <- rbind(alldata, c(widths[i], widths[i+1], heights[j, i], heights[j+1, i]));
    }
  }
  colnames(alldata) <- c("xmin", "xmax", "ymin", "ymax")

  alldata[[xvar]] <- rep(dimnames(mytable)[[1]],rep(ncol(mytable), nrow(mytable)));
  alldata[[yvar]] <- rep(dimnames(mytable)[[2]],nrow(mytable));

  ggplot(alldata, aes(xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax)) + 
    geom_rect(color="black", aes_string(fill=yvar)) +
    xlab(paste(xvar, "(count)")) + ylab(paste(yvar, "(proportion)"));
}

Пример:

makeplot_mosaic(mtcars, vs, gear)

Answer 4

Вы можете использовать пакет расширений ggplot2 под названием "ggmosaic" ( https://github.com/haleyjeppson/ggmosaic).

Обширный учебник с примером кода и визуальными результатами приведен здесь https://cran.r-project.org/web/packages/ggmosaic/vignettes/ggmosaic.html.

Answer 5

Plotluck - это библиотека, основанная на ggplot2, которая предназначена для автоматизации выбора типа графика на основе характеристик 1-3 переменных. Содержит функцию для мозаичных сюжетов. Пример:plotluck(mtcars,vs,gear)

Answer 6

Следуя предложению пользователя user2030503, вот версия, в которой используетсяggmosaic. (Обратите внимание, что ggplot 3.0 сломал часть ggmosaic, поэтому вам нужна самая последняя версия.)

library(tidyverse)
library(ggmosaic)

# Data copied from linked blog post
df <- data.frame(
  segment = LETTERS[1:4],
  segpct = c(40, 30, 20, 10),
  Alpha = c(60, 40, 30, 25), 
  Beta = c(25, 30, 30, 25),
  Gamma = c(10, 20, 20, 25), 
  Delta = c(5, 10, 20, 25)
  )

# Convert to "long" for plotting
df_long <- gather(df, key = "greek_letter", value = "pct", 
                  -c("segment", "segpct")) %>% 
  mutate(
    greek_letter = factor(
      greek_letter, 
      levels = c("Alpha", "Beta", "Gamma", "Delta")
      ),
    weight = (segpct * pct) / 10000
    )

# Plot
ggplot(df_long) +
  geom_mosaic(aes(x = product(greek_letter, segment), fill = greek_letter,
                  weight = weight))

Answer 7

Спасибо всем, кто создал эту запись, которая действительно помогла мне, так как ggmosaic не делал то, что я хотел (и не маркировал оси должным образом). Хорошая функция от Z.Lin выдает предупреждение, объясненное в https://github.com/tidyverse/ggplot2/issues/3142, которое, кажется, говорит, что предупреждение, технически неверное по своему содержанию, действительно предупреждает нас, что ggplotocracy, благослови их и поблагодарить их, считают, что geom_bar не должен иметь переменной ширины. Думаю, я понял смысл, поэтому я воспользовался функцией Джейка Фишера и настроил ее под свои нужды. Если это будет полезно другим, вот оно:

makeplot_mosaic2 <- function(data, x, y, statDigits = 1, residDigits = 1, pDigits = 3, ...){
  ### from https://stackru.com/questions/19233365/how-to-create-a-marimekko-mosaic-plot-in-ggplot2,
  ### this from Jake Fisher (I think)
  xvar <- deparse(substitute(x))
  yvar <- deparse(substitute(y))
  mydata <- data[c(xvar, yvar)]
  mytable <- table(mydata)
  
  widths <- c(0, cumsum(apply(mytable, 1, sum)))
  heights <- apply(mytable, 1, function(x){c(0, cumsum(x/sum(x)))})
  
  alldata <- data.frame()
  allnames <- data.frame()
  for(i in 1:nrow(mytable)){
    for(j in 1:ncol(mytable)){
      alldata <- rbind(alldata, c(widths[i], widths[i+1], heights[j, i], heights[j+1, i]))
    }
  }
  colnames(alldata) <- c("xmin", "xmax", "ymin", "ymax")
  
  alldata[[xvar]] <- rep(dimnames(mytable)[[1]],rep(ncol(mytable), nrow(mytable)))
  alldata[[yvar]] <- rep(dimnames(mytable)[[2]],nrow(mytable))
  
  chisq <- chisq.test(mytable)
  df <- chisq$parameter
  pval <- chisq$p.value
  chisqval <- chisq$statistic
  # stdResids <- chisq$stdres
  alldata$xcent <- (alldata$xmin + alldata$xmax)/2
  alldata$ycent <- (alldata$ymin + alldata$ymax)/2
  alldata$stdres <- round(as.vector(t(chisq$stdres)), residDigits)
  # print(chisq$stdres)
  # print(alldata)
  
  titleTxt1 <- paste0("Mosaic plot of ",
                      yvar,
                      " against ",
                      xvar,
                      ", ")
  titleTxt2 <- paste0("chisq(",
                     df,
                     ") = ",
                     round(chisqval, statDigits),
                     ", p = ",
                     format.pval(pval, digits = pDigits))
  titleTxt <- paste0(titleTxt1, titleTxt2)
  subTitleTxt <- "Cell labels are standardised residuals"
  
  ggplot(data  = alldata, 
         aes(xmin=xmin, xmax=xmax, ymin=ymin, ymax=ymax)) + 
    geom_rect(color="black", aes_string(fill=yvar)) +
    geom_text(aes(x = xcent, y = ycent, label = stdres)) +
    xlab(paste0("Count of '", 
                xvar,
                "', total = ",
                max(alldata$xmax))) + # tweaked by CE
    ylab(paste0("Proportion of '", 
                yvar,
                "' per level of '",
                xvar,
                "'")) +
    ggtitle(titleTxt,
            subtitle = subTitleTxt) +
    theme_bw() +
    theme(plot.title = element_text(hjust = .5),
          plot.subtitle = element_text(hjust = .5))
}

makeplot_mosaic2(mtcars, vs, gear)
makeplot_mosaic2(diamonds, cut, clarity)