r - 如何在 geom_split_violin 中设置不同的宽度值?
问题描述
众所周知,可以使用geom_split_violin
( Split violin plot with ggplot2 ) 绘制分割小提琴图。
# Package
library("ggplot2")
# Function
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, ...))
}
# Data
set.seed(20160229)
my_data = data.frame(
y=c(rnorm(1000), rnorm(1000, 0.5), rnorm(1000, 1), rnorm(1000, 1.5)),
x=c(rep('a', 2000), rep('b', 2000)),
m=c(rep('i', 1000), rep('j', 2000), rep('i', 1000))
)
# Figure1
ggplot(my_data, aes(x, y, fill = m)) + geom_split_violin(width = 1)
在这里,我想分别为左半小提琴和右半小提琴设置不同的宽度值(例如,0.2 和 1)。这是我的尝试:
# Figure 2
ggplot(my_data, aes(x, y, fill = m)) + geom_split_violin(width = c(0.2, 1))
希望左半部小提琴比图 1 中的小,右半部小提琴与图 1 中的相同。
解决方案
我在下面的代码中添加了一个新参数,作为现有密度高度的相对比例乘数:
# Function
GeomSplitViolin <- ggproto(
"GeomSplitViolin", GeomViolin,
draw_group = function(self, data, rel.scale, ..., draw_quantiles = NULL) {
grp <- data[1, "group"]
rel.scale <- rel.scale / max(rel.scale) # rescale to (0, 1] range
rel.scale <- rel.scale[ifelse(grp %% 2 == 1, 1, 2)] # keep only first OR second part of relative scale
data <- transform(data,
xminv = x - violinwidth * (x - xmin) * rel.scale, # apply scale multiplier
xmaxv = x + violinwidth * (xmax - x) * rel.scale)
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, rel.scale = c(1, 1)) {
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,
rel.scale = rel.scale, ...))
}
用法(仅为第一个图显示的图像):
ggplot(my_data, aes(x, y, fill = m)) +
geom_split_violin(rel.scale = c(0.2, 1))
# equivalent to above
ggplot(my_data, aes(x, y, fill = m)) +
geom_split_violin(rel.scale = c(1, 5))
# multipler can be applied on top of existing scale / width parameters
ggplot(my_data, aes(x, y, fill = m)) +
geom_split_violin(scale = "count", rel.scale = c(1, 5))
ggplot(my_data, aes(x, y, fill = m)) +
geom_split_violin(width = 0.5, rel.scale = c(1, 5))
推荐阅读
- java - 如何使用 Java 将 3*3 矩阵转换为以下格式的 9*3 矩阵?
- jmeter - 在 Jmeter 上使用 perfmon - 通过进程名称监控问题
- java - 是否可以保存一些列表,每个列表都在 CSV 文件的另一列中?JAVA
- javascript - Javascript调用其他函数
- html - 如果没有硬刷新,html 更改不会反映
- android - 设置默认语言环境 LTR (Android)
- android - 颤动中的可移动容器
- c++ - 为什么在这种情况下“if constexpr”的行为不符合预期?
- java - 如何在一个项目中同时使用 Cassandra 和 MYSQL?
- arrays - 嵌套 *ngFors - 更好的选择?(角度 7)