ggplot2 函数 - 检查用户输入变量是否应该是映射美学

Question

我编写了一个函数来制作散点图，该函数允许用户输入点的大小作为数值（保留在 aes() 调用之外）或作为数据框中的变量被映射（需要进入aes() 调用）。我远不是 NSE 方面的专家，虽然我已经开始工作了，但我觉得一定有更好的方法来做到这一点？

函数的简化版如下：

library(tidyverse)

data <- tibble(x = 1:10, y = 1:10)

test_func <- function(data, variable = 6){
  # capture the variable in vars (I think quote would also work in this function)
  variable <- vars({{variable}})
  
  # convert it to a string and check if the string starts with a digit
  # i.e. checking if this is a simple point size declaration not an aes mapping
  is_number <- variable[[1]] %>% 
    rlang::as_label() %>% 
    str_detect("^[:digit:]*$")
  
  # make initial ggplot object
  p <- ggplot(data, aes(x = x, y = y))
  
  # if variable is a simple number, add geom_point with no aes mapping
  if(is_number){
    variable <- variable[[1]] %>% 
      rlang::as_label() %>% 
      as.numeric()
    
    p <- p + geom_point(size = variable)
  } else{
    # otherwise it must be intended as an aes mapping variable  
    variable <- variable[[1]] %>% 
      rlang::as_label()
    
    p <- p + geom_point(aes(size = .data[[variable]]))
  }
  p
}

# works as a number
test_func(data, 10)

# works as a variable
test_func(data, y)

^{由reprex package (v2.0.0) 于 2021-04-08 创建}

Answer 1

一个选项是检查用户提供给variable 的内容是否是data 中的一列。如果是，请在 aes() 映射中使用该列。如果没有，请评估变量并将结果提供给aes() 之外的size：

test_func <- function(data, variable = 6) {
  v <- enquo(variable)
  gg <- ggplot(data, aes(x=x, y=y))
  
  if(exists(rlang::quo_text(v), data))
    gg + geom_point(aes(size=!!v))
  else
    gg + geom_point(size = rlang::eval_tidy(v))
}

# All of these work as expected:
test_func(data)      # Small points
test_func(data, 10)  # Bigger points
test_func(data, x)   # Using column x

作为奖励，此解决方案允许您传递存储在变量中的值，而不是直接输入数字。只要变量名不在data 中，它就会被正确评估并馈送到size=：

z <- 20
test_func(data, z)   # Works

Answer 2

你是对的。 NSE 可能有点麻烦。但是，如果您使用is.object()，请与 NSE 打交道，然后在您的 geom_point 通话中使用 show.legend 进行一些欺骗......

test_func <- function(data, variable = 6){
  qVariable = enquo(variable)
  needLegend <- !is.object(qVariable)
  data %>% 
    ggplot(aes(x = x, y = y)) + 
    geom_point(
      aes(size = !! qVariable), 
      show.legend=needLegend
    )
}

在两个测试用例中提供与函数相同的输出。