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

Question

这个问题已经有答案了:

已关闭 12 年前。

Possible Duplicate:
Pass a data.frame column name to a function

我正在尝试在 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