R 中的函数，传递数据框和列名

Question

可能的重复：
将 data.frame 列名称传递给函数

我正在尝试在 R 中创建一个函数，其中输入之间有数据帧和列名。代码如下：

DT_CAP_COLUMN  <- function(input_table,output_table,column_name,
                           cap_function,Parameter){
  input_table$column_name
  (...)
  return(1)
}

输出：

DT_CAP_COLUMN(churn_3,churn_4,'VOICE_REVENUE','STD',3)
input_table$column_name is NA

我认为问题是 input_table$column_name 无法识别。 input_table 是 churn_3，但 input_table$column_name 返回 column_name not found。

无论如何，是否可以在不必使用传递引用包或将环境作为变量传递的情况下执行此操作？

Answer 1

您可以使用方括号索引间接引用 data.frame 中的列：

示例数据：

 dat <- data.frame(
     a = letters[1:3],
     b = LETTERS[4:6],
     c = 7:9
 )

函数：

 my.function <- function(data, col){
   data[, col]
 }

结果：

>  my.function(dat, "b" )
  b
1 D
2 E
3 F
>  my.function(dat, "c" )
  c
1 7
2 8
3 9