ggplot2带有条件构面和整洁评估的构面网格

时间:2018-11-09 00:28:17

标签: r ggplot2 tidyverse tidyeval

我想创建一个生成ggplot图并为facet_grid()的构面变量提供可选参数的函数。

特别是,如果可能的话,我想在内部包含条件逻辑 facet_grid。我也想使用整洁的评估框架-所以没有公式字符串!

但是,我所有的尝试都失败了。

library(tidyverse)
iris <- iris %>% add_column(idx = rep(1:2, 75))

我的第一次尝试失败,因为facet_grid试图找到名为NULL的变量(带有反引号)。

plot_iris <- function(df_in, facet_var = NULL){
  ggplot(df_in) +
    geom_point(aes(Sepal.Length, Sepal.Width)) +
    facet_grid(vars(!!enquo(facet_var)), vars(idx))
}

plot_iris(iris)

#> Error: At least one layer must contain all faceting variables: `NULL`.
#> * Plot is missing `NULL`
#> * Layer 1 is missing `NULL`

运行plot_iris(iris, Species)可以正常工作。

我的第二次尝试也失败了,但是出现了不同的错误消息。

plot_iris2 <- function(df_in, facet_var = NULL){
  facet_quo <- enquo(facet_var)

  ggplot(df_in) +
    geom_point(aes(Sepal.Length, Sepal.Width)) +
    facet_grid(rows = ifelse(identical(facet_quo, quo(NULL)), NULL,
                             vars(!!facet_quo)),
               cols = vars(idx))
}

plot_iris2(iris)

#> Error in ans[test & ok] <- rep(yes, length.out = length(ans))[test & ok] : 
#>   replacement has length zero
#> In addition: Warning message:
#> In rep(yes, length.out = length(ans)) :

使用非NULL可选参数运行此尝试也会失败:

plot_iris2(iris, Species)

#> Error: `rows` must be `NULL` or a `vars()` list if `cols` is a `vars()` list

我的第三次尝试也失败了,错误消息相同,但警告不同:

plot_iris3 <- function(df_in, facet_var = NULL){
  facet_quo <- enquo(facet_var)

  ggplot(df_in) +
    geom_point(aes(Sepal.Length, Sepal.Width)) +
    facet_grid(rows = vars(ifelse(identical(facet_quo, quo(NULL)), NULL,
                                  !!facet_quo)))
}

plot_iris3(iris)

#> Error in ans[test & ok] <- rep(yes, length.out = length(ans))[test & ok] : 
#>   replacement has length zero
#> In addition: Warning message:
#> In rep(yes, length.out = length(ans)) :
#>   'x' is NULL so the result will be NULL

使用非NULL可选参数将返回以idx而非Species组成的图-单行有一个构面标签,由“ 1”标记。

plot_iris3(iris, Species)

有没有其他选择可以在单个facet_grid调用中使用条件逻辑,并且在可选参数为NULL时有效吗?

2 个答案:

答案 0 :(得分:3)

也许我们应该从vars规范中删除NULL元素以使其更容易。我已经打开一个问题:https://github.com/tidyverse/ggplot2/issues/2986

您可以使用rlang::quo_is_null()检查quo(NULL)。为了清楚起见,我将在单独的步骤中进行操作。

plot_iris <- function(df_in, facet_var = NULL){
  facet_quo <- enquo(facet_var)

  if (rlang::quo_is_null(facet_quo)) {
    rows <- vars()
  } else {
    rows <- vars(!!facet_quo)
  }

  ggplot(df_in) +
    geom_point(aes(Sepal.Length, Sepal.Width)) +
    facet_grid(rows, vars(idx))
}

答案 1 :(得分:2)

我认为您需要第二种方法。关键是您需要switch()才能返回NULL,而不是ifelse()。参见示例here和讨论here

plot_iris2 <- function(df_in, facet_var = NULL){
    facet_quo <- enquo(facet_var)

    ggplot(df_in) +
        geom_point(aes(Sepal.Length, Sepal.Width)) +
        facet_grid(rows = switch(identical(facet_quo, quo(NULL)) + 1, 
                                 vars(!!facet_quo),
                                 NULL),
                   cols = vars(idx))
}

plot_iris2(iris, facet_var = NULL)

enter image description here

plot_iris2(iris, facet_var = Species)

enter image description here