如何编写ggplot' +' -pipeable函数,可以参考输入图

时间:2017-12-13 17:13:23

标签: r ggplot2 ggproto

我试图编写一个可以使用基于' -based的ggplot2语法调用的函数。

myplot + myfunction

具体来说,写入的函数将y轴对称为零,因此需要确定输入图的y轴范围。

所以,让我们,

ylim_sym <- function(p){
    get_y_range <- function(p){
        ggplot2::ggplot_build(p)$layout$panel_ranges[[1]]$y.range
        }
    max_offset <- max(abs(get_y_range(p)))
    p + ylim(- max_offset, max_offset)
}

使用此功能,以下工作:

qplot(x = 1:10, y = exp(rnorm(10))) %>% ylim_sym()

但由于+.gg%>%之间存在一些优先问题,因此无法正常工作:

qplot(x = 1:10, y = exp(rnorm(10))) +
    geom_abline(slope = 0) %>%
    ylim_sym()

(我可以写后者(all_my_ggplot_pipeline) %>% ylim_sym(),但它的语法非常难看。)

理想情况下,我希望能够写ylim_sym,以便可以像这样管道传输,

qplot(x = 1:10, y = exp(rnorm(10))) + ylim_sym()

但是我无法在+

中查看如何访问ylim_sym的LHS上的情节

有什么想法吗?

2 个答案:

答案 0 :(得分:5)

我能够通过以下方式解决它。

StatSymYLim <- ggproto(
  "StatSymYLim", Stat, 
  compute_group = function(data, scales) {
    out <- data.frame(
      x = median(data$x),
      y = c(-1, 1) * max(abs(data$y))
      )
    out
    },
    required_aes = c("x", "y")
  )

ylim_sym <- function(...){
  geom_blank(..., stat = StatSymYLim)
  }

然后根据需要进行以下工作:

qplot(x = 1:10, y = exp(rnorm(10))) +
  geom_abline(slope = 0) +
  ylim_sym()

我对ggplot2内部的理解非常不公平,所以这可能是一个天真的解决方案。

答案 1 :(得分:0)

注意:由于对象的结构略有变化,您的函数需要更新

使用package ggfun可以起作用:

# devtools::install_github("moodymudskipper/ggfun")
library(ggfun)

ylim_sym <- function(p){
  get_y_range <- function(p){
    ggplot2::ggplot_build(p)$layout$panel_params[[1]]$y.range
  }
  max_offset <- max(abs(get_y_range(p)))
  p + ylim(- max_offset, max_offset)
}

qplot(x = 1:10, y = exp(rnorm(10))) +
  geom_abline(slope = 0) +
  ylim_sym