创建一个函数以从数据集生成表

Question

我有一个数据集（我们称他为 db_data），我想从该数据集的 2 个变量中创建一个表（我们称它们为 var1 和 var2）。

db_data <- base::data.frame(var1 = c("M", "M", "F", "F", "F"), 
                            var2 = c("brown", "blue", "brown", "brown", "brown"))

我可以很容易地做到这一点：

tab <- base::table(db_data[j = c("var1", "var2")])
tab

我的问题是我想在更大的工作中多次使用它。所以我尝试创建一个函数来做到这一点，但根本没有结果。

# Create function
tab_function = function(data,
                         var1,
                         var2) {
tab <- base::table(data[j = c(var1, var2)])
tab
}

# Test
tab_function(data = db_data, var1 = "var1", var2 = "var2")

这播放没有错误，但我得到了完全不同的结果，并且没有任何对象。我尝试使用get()、noquote() 以及

# This
tab_function(data = data, var1 = var1, var2 = var2)

# This
tab_function = function(data,
                         var1,
                         var2) {
tab <- base::table(data[j = c("var1", "var2")])
tab
}

我不明白。这似乎很容易，但它不起作用。

Answer 1

您可以使用return 从函数中返回某些内容。另外，我通过更改 data.frame 中两列的选择方法来避免提到的警告 r2evans。

tab_function <- function(db_data, var1, var2){
  tab <- base::table(db_data[,names(db_data) %in% c(var1, var2)])
  return(tab)
}

tab_function(db_data, "var1", "var2")

    var2
var1 blue brown
   F    0     3
   M    1     1