如何在ggplot2(R)中使用分割小提琴图来避免重叠标签

时间:2019-08-11 11:24:26

标签: r ggplot2 label overlap violin-plot

我正在使用Split violin plot with ggplot2处创建的geom_split_violin函数生成小提琴图。

然后,我为每个分割小提琴添加样本大小的标签(n = ...)。但是,不幸的是标签重叠。我如何才能将它们稍微向左和向右移动,以免它们重叠?

这是我正在使用的完整代码,其下是带有重叠的“ n = ...”标签的结果。

 # Create data
 set.seed(20160229)
 my_data = data.frame(
     y=c(rnorm(500), rnorm(300, 0.5), rnorm(400, 1), rnorm(200, 1.5)),
     x=c(rep('a', 800), rep('b', 600)),
     m=c(rep('i', 300), rep('j', 700), rep('i', 400)))
 # Code to create geom_split_violin function from link above
 library('ggplot2')
 GeomSplitViolin <- ggproto("GeomSplitViolin", GeomViolin, 
                       draw_group = function(self, data, ..., draw_quantiles = NULL) {
    data <- transform(data, xminv = x - violinwidth * (x - xmin), xmaxv = x + violinwidth * (xmax - x))
   grp <- data[1, "group"]
   newdata <- plyr::arrange(transform(data, x = if (grp %% 2 == 1) xminv else xmaxv), if (grp %% 2 == 1) y else -y)
   newdata <- rbind(newdata[1, ], newdata, newdata[nrow(newdata), ], newdata[1, ])
   newdata[c(1, nrow(newdata) - 1, nrow(newdata)), "x"] <- round(newdata[1, "x"])
   if (length(draw_quantiles) > 0 & !scales::zero_range(range(data$y))) {
     stopifnot(all(draw_quantiles >= 0), all(draw_quantiles <=
       1))
     quantiles <- ggplot2:::create_quantile_segment_frame(data, draw_quantiles)
     aesthetics <- data[rep(1, nrow(quantiles)), setdiff(names(data), c("x", "y")), drop = FALSE]
     aesthetics$alpha <- rep(1, nrow(quantiles))
     both <- cbind(quantiles, aesthetics)
     quantile_grob <- GeomPath$draw_panel(both, ...)
     ggplot2:::ggname("geom_split_violin", grid::grobTree(GeomPolygon$draw_panel(newdata, ...), quantile_grob))
   }
   else {
     ggplot2:::ggname("geom_split_violin", GeomPolygon$draw_panel(newdata, ...))
   }
 })
 geom_split_violin <- function(mapping = NULL, data = NULL, stat = "ydensity", position = "identity", ..., 
                               draw_quantiles = NULL, trim = TRUE, scale = "area", na.rm = FALSE, 
                               show.legend = NA, inherit.aes = TRUE) {
   layer(data = data, mapping = mapping, stat = stat, geom = GeomSplitViolin, 
         position = position, show.legend = show.legend, inherit.aes = inherit.aes, 
         params = list(trim = trim, scale = scale, draw_quantiles = draw_quantiles, na.rm = na.rm, ...))
 }
 # Add labels 'n = ...'
 give_n = function(x, y_lo = min(my_data$y)) {
      data.frame(y = y_lo * 1.06,
              label = paste("n =", length(x)))
 }
 # Plot data
 ggplot(my_data, aes(x, y, fill = m)) + 
      geom_split_violin() + 
      stat_summary(fun.data = give_n, aes(x = as.factor(x)), geom = "text")

结果(注意重叠的'n = ...'标签): enter image description here

1 个答案:

答案 0 :(得分:2)

添加position_nudge()是否可以解决您的问题?

ggplot(my_data, aes(x, y, fill = m)) + 
  geom_split_violin() + 
  stat_summary(fun.data = give_n, aes(x = as.factor(x)), geom = "text",
               position = position_nudge(x = c(-0.25, 0.25)))

enter image description here