Question

我想将列名称作为变量传递给函数。具体来说，给出以下代码：

#------------------------------------------
# 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 FAQ中的文档，但我无法使其正常工作。非常感谢解决方案和简要说明。

提前致谢。

Answer 1

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简化这个，然后再分配名字：

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

Answer 2

我很困惑，你的函数的两个参数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
}

用R中的字符串值替换变量

2 个答案: