1

我想将列名的名称作为变量传递给函数。具体来说,给定以下代码:

#------------------------------------------
# Create mock data
#------------------------------------------

DF1 = data.frame(colA = as.Date("2015-09-21") + 0:4, B = 1:5, C = 5:1)
DF2 = data.frame(B = 1:5, C = 5:1)

#------------------------------------------
# Sample function call
#------------------------------------------

colname <- "colA"

b_func <- function( my_dt1, my_dt2, my_col ) {    
  return( eval(substitute( cbind( variable = as.Date("2015-09-21") + 0:4, DF2 ), 
    list(variable = as.name(my_col)))) )
}

b_func( DF1, DF2, colname )

我希望得到以下输出:

        colA B C
1 2015-09-21 1 5
2 2015-09-22 2 4
3 2015-09-23 3 3
4 2015-09-24 4 2
5 2015-09-25 5 1

相反,我得到以下输出而不是"colA"得到"variable".

    variable B C
1 2015-09-21 1 5
2 2015-09-22 2 4
3 2015-09-23 3 3
4 2015-09-24 4 2
5 2015-09-25 5 1

进一步挖掘我发现:

colname <- "colA"
substitute( cbind( variable = as.Date("2015-09-21") + 0:4, DF2 ), list(variable = as.name(colname)))
# returns:  cbind(variable = as.Date("2015-09-21") + 0:4, DF2)
# expected: cbind(colA = as.Date("2015-09-21") + 0:4, DF2)

此示例基于R 常见问题解答中的文档,但我无法使其工作。一个解决方案和简短的解释将不胜感激。

提前致谢。

4

2 回答 2

2

substitute不替换不是变量的参数名称。

您可以do.call改用:

b_func <- function (my_dt1, my_dt2, my_col)
    do.call(cbind, setNames(c(list(as.Date("2015-09-21") + 0:4), DF2),
                            c(my_col, colnames(DF2))))

但理查德在他的评论中有一点:你可以通过cbind先 ing 来简化这一点,然后再分配名称:

b_func <- function (my_dt1, my_dt2, my_col)
    setNames(cbind(as.Date("2015-09-21") + 0:4, DF2), c(my_col, colnames(DF2)))
于 2015-09-23T22:01:29.567 回答
1

我对您的函数的两个参数 my_dt1 和 my_dt2 似乎没有使用感到困惑。您似乎正在做的只是向 DF2 添加一列?如果是这样,只需执行以下操作:

b_func = function(my_dt, my_col) {
  my_dt[[my_col]] = as.Date("2015-09-21") + 0:4
  my_dt
}
于 2015-09-23T22:40:57.150 回答