列参考 data.table function R

Question

我正在尝试制作一个 function 来调用作为以下 arguments 之一提供的数据表中的列：

df <- read.table(text = "x1 x2 y
CA 20 50
CA 30.5 100
CA 40.5 200
AZ 20.12 400
AZ 25 500
OR 86 600
OR 75 700
OR 45 800", header = TRUE)

df$x1 <- as.factor(df$x1)

library(data.table)

make_freq <- function(df, var_name){
  
  df <- df 
  setDT(df)
  
  
  tb <- df[, .N, by = var_name][,prop_ := round(((N/sum(N))*100), digits = 0)][order(var_name)]
  
  gg1 <- ggplot(tb, aes(x = var_name, y = prop_)) +
    geom_bar(width = .35, stat = "identity", color = "darkblue", fill = "darkblue") +
    ggtitle(paste0("var_name")) +
    theme_bw() +
    theme(plot.title = element_text(size = 10)) +
    theme(axis.text.x = element_text(angle = 45)) 
  
  return(list(figure = gg1))
}

make_freq(df = df, var_name = x1)

理想情况下，我希望能够运行 function 以便我可以使用var_name参数为我想要的任何分类变量创建 ggplot 图。 我收到Object x1 not found错误，这让我认为我需要引用或取消引用 function 或其他内容中的var_name参数。

Answer 1

是的，如果您想使用非标准评估，您需要引用var_name参数。 只需添加：

var_name <- substitute(var_name)

到 function 的顶部。 请注意，在这种情况下，默认的 x 轴 label 将是var_name 。 如果您希望它默认为作为var_name传递的任何内容，则需要执行几个额外的步骤。 将 function 的顶部更改为：

  x <- enquo(var_name)
  var_name <- substitute(var_name)

然后修改tb行。

  tb <- df[, .N, by = eval(deparse(var_name))][,prop_ := round(((N/sum(N))*100), digits = 0)][order(eval(var_name))]

然后在ggplot() ：

gg1 <- ggplot(tb, aes(x = !!x, y = prop_)) + ...

Answer 2

1. 你应该引用x1因为你没有这个 object （这是列的名称）。
2. data.table object 中的参数by可能是字符，而df[, .N, by = var_name]是好的代码。 但是[order(var_name)]是错误的。 您可以使用[order(get(var_name))] 。
3. 因为var_name是字符，我们需要在 ggplot 中将var_name更改为get(var_name) 。

完整代码：

make_freq <- function(df, var_name){
    
    df <- df 
    setDT(df)
    
    
    tb <- df[, .N, by = var_name][,prop_ := round(((N/sum(N))*100), digits = 0)][order(get(var_name))]
    
    gg1 <- ggplot(tb, aes(x = get(var_name), y = prop_)) +
        geom_bar(width = .35, stat = "identity", color = "darkblue", fill = "darkblue") +
        ggtitle(paste0("var_name")) +
        theme_bw() +
        theme(plot.title = element_text(size = 10)) +
        theme(axis.text.x = element_text(angle = 45)) 
    
    return(list(figure = gg1))
}


make_freq(df = df, var_name = "x1")