R ggplot2:将样本大小信息添加到x轴刻度标签
这个问题与 (注意:所有功能都已简化;没有对正确的对象类型、NAs等进行错误检查。) 在base R中,很容易创建一个函数来生成一个条形图,其样本大小显示在分组变量的每个级别下面:您可以使用R ggplot2:将样本大小信息添加到x轴刻度标签,r,ggplot2,ggproto,R,Ggplot2,Ggproto,这个问题与 (注意:所有功能都已简化;没有对正确的对象类型、NAs等进行错误检查。) 在base R中,很容易创建一个函数来生成一个条形图,其样本大小显示在分组变量的每个级别下面:您可以使用mtext()函数添加样本大小信息: stripchart_w_n_ver1 <- function(data, x.var, y.var) { x <- factor(data[, x.var]) y <- data[, y.var] # Need to call plot
mtext()
函数添加样本大小信息:
stripchart_w_n_ver1 <- function(data, x.var, y.var) {
x <- factor(data[, x.var])
y <- data[, y.var]
# Need to call plot.default() instead of plot because
# plot() produces boxplots when x is a factor.
plot.default(x, y, xaxt = "n", xlab = x.var, ylab = y.var)
levels.x <- levels(x)
x.ticks <- 1:length(levels(x))
axis(1, at = x.ticks, labels = levels.x)
n <- sapply(split(y, x), length)
mtext(paste0("N=", n), side = 1, line = 2, at = x.ticks)
}
stripchart_w_n_ver1(mtcars, "cyl", "mpg")
stripchart_w_n_ver2 <- function(data, x.var, y.var) {
x <- factor(data[, x.var])
y <- data[, y.var]
# Need to set the second element of mgp to 1.5
# to allow room for two lines for the x-axis tick labels.
o.par <- par(mgp = c(3, 1.5, 0))
on.exit(par(o.par))
# Need to call plot.default() instead of plot because
# plot() produces boxplots when x is a factor.
plot.default(x, y, xaxt = "n", xlab = x.var, ylab = y.var)
n <- sapply(split(y, x), length)
levels.x <- levels(x)
axis(1, at = 1:length(levels.x), labels = paste0(levels.x, "\nN=", n))
}
stripchart_w_n_ver2(mtcars, "cyl", "mpg")
虽然在base R中这是一项非常简单的任务,但在ggplot2中却极其复杂,因为很难获取用于生成绘图的数据,而且虽然有相当于axis()
(例如scale\u x_discrete
等)的函数,但没有相当于mtext()的函数
,使您可以轻松地将文本放置在页边距内的指定坐标处
我尝试使用内置的stat\u summary()
函数来计算样本大小(即fun.y=“length”
),然后将该信息放在x轴刻度标签上,但据我所知,您无法提取样本大小,然后使用scale\u discrete()函数以某种方式将其添加到x轴刻度标签上
,您必须告诉stat\u summary()
您希望它使用什么geom。您可以设置geom=“text”
,但是您必须提供标签,重点是标签应该是样本大小的值,这是stat\u summary()
正在计算的,但您无法得到(您还必须指定文本的放置位置,同样,很难确定文本的放置位置,使其直接位于x轴刻度标签的下方)
小插曲“扩展ggplot2”()向您展示了如何创建自己的统计函数,允许您直接获取数据,但问题是您始终必须定义一个几何图形才能与统计函数配合使用(即,ggplot
认为您希望在绘图中绘制此信息,而不是在边距中);据我所知,您不能在自定义stat函数中获取计算的信息,也不能在绘图区域中绘制任何内容,而是将信息传递给scale函数,如scale\x\u discrete()
。我试着这样做;我所能做的就是将每组的样本大小信息设置为最小值y:
StatN <- ggproto("StatN", Stat,
required_aes = c("x", "y"),
compute_group = function(data, scales) {
y <- data$y
y <- y[!is.na(y)]
n <- length(y)
data.frame(x = data$x[1], y = min(y), label = paste0("n=", n))
}
)
stat_n <- function(mapping = NULL, data = NULL, geom = "text",
position = "identity", inherit.aes = TRUE, show.legend = NA,
na.rm = FALSE, ...) {
ggplot2::layer(stat = StatN, mapping = mapping, data = data, geom = geom,
position = position, inherit.aes = inherit.aes, show.legend = show.legend,
params = list(na.rm = na.rm, ...))
}
ggplot(mtcars, aes(x = factor(cyl), y = mpg)) + geom_point() + stat_n()
但是,此功能不能正确处理镶嵌面。例如:
ggstripchart(mtcars, "cyl", "mpg") + facet_wrap(~am)
pgg = function(dat, x, y, facet_r=NULL, facet_c=NULL) {
# Convert x-variable to a factor
dat[,x] = as.factor(dat[,x])
# Plot points
p = ggplot(dat, aes_string(x, y)) +
geom_point(position=position_jitter(w=0.3, h=0)) + theme_bw()
# Summarise data to get counts by x-variable and (if present) facet variables
dots = lapply(c(facet_r, facet_c, x), as.symbol)
nn = dat %>% group_by_(.dots=dots) %>% tally
# If there are facets, add them to the plot
if (!is.null(facet_r) | !is.null(facet_c)) {
facets = paste(ifelse(is.null(facet_r),".",facet_r), " ~ " ,
ifelse(is.null(facet_c),".",facet_c))
p = p + facet_grid(facets)
}
# Add counts as text labels
p + geom_text(data=nn, aes(label=paste0("N = ", nn$n)),
y=min(dat[,y]) - 0.15*min(dat[,y]), colour="grey20", size=3) +
scale_y_continuous(limits=range(dat[,y]) + c(-0.1*min(dat[,y]), 0.01*max(dat[,y])))
}
pgg(mtcars, "cyl", "mpg")
pgg(mtcars, "cyl", "mpg", facet_c="am")
pgg(mtcars, "cyl", "mpg", facet_c="am", facet_r="vs")
显示了为每个方面组合的两个方面的样本大小。我必须将方面构建到包装器函数中,这样就无法尝试使用ggplot
提供的所有功能
如果有人对此问题有任何见解,我将不胜感激。非常感谢您抽出时间!我已更新了
包中包含一个名为stat\n\u text
的stat\n\u text
,它将在每个唯一的x值下面添加样本大小(唯一y值的数量)。有关更多信息和示例列表,请参阅forstat\n\u text
。下面是一个简单的示例:
library(ggplot2)
library(EnvStats)
p <- ggplot(mtcars,
aes(x = factor(cyl), y = mpg, color = factor(cyl))) +
theme(legend.position = "none")
p + geom_point() +
stat_n_text() +
labs(x = "Number of Cylinders", y = "Miles per Gallon")
库(ggplot2)
图书馆(环境统计)
p如果关闭剪裁,您可以使用geom_text
打印x轴标签下方的计数,但可能需要调整位置。我在下面的代码中包含了一个“微移”参数。此外,下面的方法适用于所有面(如果有)都是柱面的情况
我意识到您最终想要的代码将在一个新的geom中工作,但下面的示例可能适合在geom中使用
library(ggplot2)
library(dplyr)
pgg = function(dat, x, y, facet=NULL, nudge=0.17) {
# Convert x-variable to a factor
dat[,x] = as.factor(dat[,x])
# Plot points
p = ggplot(dat, aes_string(x, y)) +
geom_point(position=position_jitter(w=0.3, h=0)) + theme_bw()
# Summarise data to get counts by x-variable and (if present) facet variables
dots = lapply(c(facet, x), as.symbol)
nn = dat %>% group_by_(.dots=dots) %>% tally
# If there are facets, add them to the plot
if (!is.null(facet)) {
p = p + facet_grid(paste("~", paste(facet, collapse="+")))
}
# Add counts as text labels
p = p + geom_text(data=nn, aes(label=paste0("N = ", nn$n)),
y=min(dat[,y]) - nudge*1.05*diff(range(dat[,y])),
colour="grey20", size=3.5) +
theme(axis.title.x=element_text(margin=unit(c(1.5,0,0,0),"lines")))
# Turn off clipping and return plot
p <- ggplot_gtable(ggplot_build(p))
p$layout$clip[p$layout$name=="panel"] <- "off"
grid.draw(p)
}
pgg(mtcars, "cyl", "mpg")
pgg(mtcars, "cyl", "mpg", facet=c("am","vs"))
我的解决方案可能有点简单,但效果很好
给出了一个使用faceting by的示例,我从创建标签开始
使用粘贴
和\n
mtcars2 <- mtcars %>%
group_by(cyl, am) %>% mutate(n = n()) %>%
mutate(label = paste0(cyl,'\nN = ',n))
生成如下图所示的内容
非常感谢您在这方面的帮助!在我发布问题后,我已经按照您的第二个建议,找到了如何将样本大小放置在绘图面板底部的方法。我几乎完成了新的统计函数和几何图形,它们将实现我想要的功能,并将它们纳入我的EnvStats软件包的下一个版本中(我这样做的时候会在这里发布)。再次感谢您的帮助和建议!嗨,史蒂夫,有没有办法删除“n=”?我只想显示数字。
mtcars2 <- mtcars %>%
group_by(cyl, am) %>% mutate(n = n()) %>%
mutate(label = paste0(cyl,'\nN = ',n))
ggplot(mtcars2,
aes(x = factor(label), y = mpg, color = factor(label))) +
geom_point() +
xlab('cyl') +
facet_wrap(~am, scales = 'free_x') +
theme(legend.position = "none")