如何为ggplot自定义主题函数添加非标准参数以控制网格线显示轴？

问题概述 #

需要编写ggplot自定义主题函数，通过my_theme(gridline_axis = "x")或my_theme(gridline_axis = "y")分别只显示X轴或Y轴的网格线，同时保留其他统一的主题设置。尝试多种条件设置方法均失败，现有代码无法达到预期效果。

修正后的代码方案 #

核心思路是根据参数明确指定需要显示的轴网格线，将另一轴的所有网格线设为element_blank()，同时保留通用主题设置，避免重复代码：

my_theme <- function(gridline_axis = "x") {
  # 基础主题
  base_theme <- theme_bw(base_size = 12)
  
  # 通用主题设置
  common_settings <- list(
    # 文本设置
    plot.title = element_text(size = 15, vjust = 3),
    axis.title = element_text(size = 13),
    axis.title.x = element_text(vjust = -2),
    axis.title.y = element_text(angle = 90, vjust = 5.5),
    # 画布设置
    plot.background = element_rect(colour = "black", linewidth = 1),
    plot.margin = unit(c(1, 1, 1, 1), "cm"),
    # 面板设置
    panel.background = element_blank(),
    panel.border = element_rect(color = "red", linewidth = 2)
  )
  
  # 根据参数生成网格线设置
  grid_settings <- if (gridline_axis == "x") {
    list(
      # 显示X轴主次网格线
      panel.grid.major.x = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      panel.grid.minor.x = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      # 隐藏Y轴所有网格线
      panel.grid.major.y = element_blank(),
      panel.grid.minor.y = element_blank()
    )
  } else if (gridline_axis == "y") {
    list(
      # 显示Y轴主次网格线
      panel.grid.major.y = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      panel.grid.minor.y = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      # 隐藏X轴所有网格线
      panel.grid.major.x = element_blank(),
      panel.grid.minor.x = element_blank()
    )
  } else {
    # 默认显示X轴网格线，或可添加错误提示
    list(
      panel.grid.major.x = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      panel.grid.minor.x = element_line(color = "gray", linewidth = 0.5, linetype = 1),
      panel.grid.major.y = element_blank(),
      panel.grid.minor.y = element_blank()
    )
  }
  
  # 合并所有设置并替换基础主题
  base_theme %+replace% do.call(theme, c(common_settings, grid_settings))
}

# 测试代码
library(ggplot2)
p <- ggplot(mtcars, aes(mpg, wt)) + geom_point()

# 只显示X轴网格线
p + my_theme(gridline_axis = "x")

# 只显示Y轴网格线
p + my_theme(gridline_axis = "y")

方案说明 #

• 通用设置与条件设置分离：将所有固定的主题配置放在common_settings中，网格线的条件配置单独放在grid_settings，避免重复代码。
• 明确覆盖网格线：直接针对对应轴的主次网格线设置显示/隐藏，彻底覆盖theme_bw默认的网格线样式，解决原代码中命名和元素顺序导致的失效问题。
• 参数容错处理：添加默认分支，当传入非"x"/"y"参数时，默认使用X轴网格线设置（也可根据需求改为报错提示）。

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