この目的のために、新しいStat関数を作成しました。
、、を引数として取りnbins
、デフォルトは4つすべてです。bin_var
bin_fun
summary_fun
- のデフォルトは
nbins
、データポイントの数によって異なります。
- のデフォルト
bin_var
は「x」です。「y」に設定することもできます。これは、に供給される変数を指定しますbin_fun
。
bin_fun
ビニング関数です。デフォルトでは、それseq_cut
は私が目的のために書いたものです。独自のビニング関数を作成することもできます。データとnbinを引数として取る必要があります。
summary_fun
ビンを集約するために使用される要約関数です。デフォルトでは、ですmean
。xとyの集計関数をとで個別に指定することもできfun.x
ますfun.y
。
- 美学として
ymin
とをとるgeomを使用する場合は、とを指定することもできます。ymax
fun.ymin
fun.ymax
aes(group = your_bins)を指定した場合、bin_fun
は無視され、代わりにグループ化変数が使用されることに注意してください。また、としてアクセスできるカウント変数が作成されることにも注意してください..count..
。
あなたの場合、あなたはそれを次のように使用します:
p <- ggplot(data, aes(x, y)) +
geom_point(aes(size = ..count..), stat = "binner") +
ylim(0, 1)
この場合はあまり有用ではありませんが(これは等分散性を示しており、Bern(0.5)変量の仮定にふさわしい分散は約0.25ですが)、例としてだけです。
p + geom_linerange(stat = "binner",
fun.ymin = function(y) mean(y) - var(y) / 2,
fun.ymax = function(y) mean(y) + var(y) / 2)
コード:
library(proto)
stat_binner <- function (mapping = NULL, data = NULL, geom = "point", position = "identity", ...) {
StatBinner$new(mapping = mapping, data = data, geom = geom, position = position, ...)
}
StatBinner <- proto(ggplot2:::Stat, {
objname <- "binner"
default_geom <- function(.) GeomPoint
required_aes <- c("x", "y")
calculate_groups <- function(., data, scales, bin_var = "x", nbins = NULL, bin_fun = seq_cut, summary_fun = mean,
fun.data = NULL, fun.y = NULL, fun.ymax = NULL, fun.ymin = NULL,
fun.x = NULL, fun.xmax = NULL, fun.xmin = NULL, na.rm = FALSE, ...) {
data <- remove_missing(data, na.rm, c("x", "y"), name = "stat_binner")
# Same rules as binnedplot in arm package
n <- nrow(data)
if (is.null(nbins)) {
nbins <- if (n >= 100) floor(sqrt(n))
else if (n > 10 & n < 100) 10
else floor(n/2)
}
if (length(unique(data$group)) == 1) {
data$group <- bin_fun(data[[bin_var]], nbins)
}
if (!missing(fun.data)) {
# User supplied function that takes complete data frame as input
fun.data <- match.fun(fun.data)
fun <- function(df, ...) {
fun.data(df$y, ...)
}
} else {
if (!is.null(summary_fun)) {
if (!is.null(fun.x)) message("fun.x overriden by summary_fun")
if (!is.null(fun.y)) message("fun.y overriden by summary_fun")
fun.x <- fun.y <- summary_fun
}
# User supplied individual vector functions
fs_x <- compact(list(xmin = fun.x, x = fun.x, xmax = fun.xmax))
fs_y <- compact(list(ymin = fun.ymin, y = fun.y, ymax = fun.ymax))
fun <- function(df, ...) {
res_x <- llply(fs_x, function(f) do.call(f, list(df$x, ...)))
res_y <- llply(fs_y, function(f) do.call(f, list(df$y, ...)))
names(res_y) <- names(fs_y)
names(res_x) <- names(fs_x)
as.data.frame(c(res_y, res_x))
}
}
summarise_by_x_and_y(data, fun, ...)
}
})
summarise_by_x_and_y <- function(data, summary, ...) {
summary <- ddply(data, "group", summary, ...)
count <- ddply(data, "group", summarize, count = length(y))
unique <- ddply(data, "group", ggplot2:::uniquecols)
unique$y <- NULL
unique$x <- NULL
res <- merge(merge(summary, unique, by = "group"), count, by = "group")
# Necessary for, eg, colour aesthetics
other_cols <- setdiff(names(data), c(names(summary), names(unique)))
if (length(other_cols) > 0) {
other <- ddply(data[, c(other_cols, "group")], "group", numcolwise(mean))
res <- merge(res, other, by = "group")
}
res
}
seq_cut <- function(x, nbins) {
bins <- seq(min(x), max(x), length.out = nbins)
findInterval(x, bins, rightmost.closed = TRUE)
}