如何仅为部分分面设置geom_density()的边界

问题场景 #

我使用ggplot2绘制分面图展示多变量分布，每个分面包含直方图和核密度估计。模拟示例代码如下：

library(tidyverse)

set.seed(42)
N <- 100

a <- rgamma(n = N, shape = 2)
b <- rnorm(n = N)

d <- tibble(x = 1:N, a = a, b = b) %>% 
    pivot_longer(cols = c("a", "b"), values_to = "Value", names_to = "Variable")

d %>% ggplot(mapping = aes(x = Value)) +
    geom_histogram(mapping = aes(y = ..density..)) + 
    geom_density() + 
    facet_wrap("Variable")

其中变量a服从伽马分布（非负），需要设置密度估计边界c(0, Inf)，但直接给geom_density()加bounds参数会应用到所有分面，无法实现分变量个性化设置。

解决方案 #

核心思路是为每个变量单独预计算符合其边界要求的密度估计数据，再用geom_line()绘制到分面图中，具体实现如下：

方法1：分组计算个性化密度数据 #

library(tidyverse)

set.seed(42)
N <- 100

a <- rgamma(n = N, shape = 2)
b <- rnorm(n = N)

d <- tibble(x = 1:N, a = a, b = b) %>% 
    pivot_longer(cols = c("a", "b"), values_to = "Value", names_to = "Variable")

# 按变量分组，计算对应边界的密度数据
density_data <- d %>%
    group_by(Variable) %>%
    group_modify(function(data, group_key) {
        # 根据变量名指定边界
        if (group_key$Variable == "a") {
            dens_obj <- density(data$Value, bounds = c(0, Inf))
        } else {
            dens_obj <- density(data$Value)
        }
        # 转换为ggplot可用的数据框格式
        tibble(x = dens_obj$x, density = dens_obj$y)
    })

# 绘制分面图：直方图+个性化密度线
d %>% ggplot(aes(x = Value)) +
    geom_histogram(aes(y = ..density..), alpha = 0.5, bins = 15) +
    geom_line(data = density_data, aes(x = x, y = density), color = "firebrick", linewidth = 1) +
    facet_wrap(~Variable)

方法2：用字典管理多变量边界（适用于变量较多场景） #

如果有多个变量需要设置不同边界，可以用字典统一管理规则，代码更易维护：

# 定义变量-边界映射字典
bounds_map <- list(
    a = c(0, Inf),
    b = NULL  # 不设置边界
    # 可添加更多变量的边界规则
)

# 分组计算密度数据
density_data <- d %>%
    group_by(Variable) %>%
    group_modify(function(data, group_key) {
        current_bounds <- bounds_map[[group_key$Variable]]
        dens_obj <- if (!is.null(current_bounds)) {
            density(data$Value, bounds = current_bounds)
        } else {
            density(data$Value)
        }
        tibble(x = dens_obj$x, density = dens_obj$y)
    })

# 绘图代码同方法1

这种方式完全规避了geom_density()全局参数的限制，每个分面的密度线都严格遵循对应变量的边界要求，同时保持分面图的整体一致性。

内容的提问来源于stack exchange，提问作者BestGirl