在函数调用中提供变量名并绑定到 R 中的现有数据框

Question

目标是让函数调用的最后一个参数为要绑定到原始数​​据框的新列提供名称。

参考this 和this 上一个问题，并以第一个问题的最小工作示例为基础。

GroupId <-          c(1,1,1,1,2,2,2,3,3)
IndId <-            c(1,1,2,2,3,4,4,5,5)
IndGroupProperty <- c(1,2,1,2,3,3,4,5,6)
PropertyType <-     c(1,2,1,2,2,2,1,2,2)

df <- data.frame(GroupId, IndId, IndGroupProperty, PropertyType)
df

ValidGroupC <-       c(1,1,1,1,0,0,0,0,0)
df <- data.frame(df, ValidGroupC)
df

library(dplyr)
grouptest <- function(object, group, ind, type, new){
groupvar <- deparse(substitute(group)) 
indvar <- deparse(substitute(ind)) 
typevar <- deparse(substitute(type)) 
eval(substitute(
tmp <- object[, c(groupvar, indvar, typevar)] %.%
  group_by(group, ind) %.%
  mutate(type1 = any(type == 1))  %.%
  group_by(group, add = FALSE) %.%
  mutate(tmp2 = all(type1) * 1) %.%
  select(-type1)
  ))
new <- tmp[, 4]                    # this is the relevant part
tmp <- cbind(object, new)          # this is the relevant part
}

df <- grouptest(df, GroupId, IndId, PropertyType, ValidGroup)
df

所以大部分代码已经是引用问题的产物。这个问题的相关部分在最后，我将我对tmp 所做的计算的第 4 列放入一个新对象中，该对象的名称应取自函数调用中的new 参数，然后我将其绑定到原始数​​据框。

我的问题：为什么最后一列 df 没有命名为 ValidGroup ？ 我不明白有什么问题 - new 应该替换为 ValidGroup，但不是吗？

我尝试将这两行放在eval() 中，结果为Error in cbind(df, ValidGroup) : object 'ValidGroup' not found。

我尝试在两条线周围加上另一个eval(substitute())，同样的错误。

我已经尝试了许多其他变体来放置线条，使用经过解析的newvar，将tmp 命名为new，。 . .

Answer 1

您想将函数中突出显示的最后两行更改为：

object[, new] <- tmp[, 4]
object

然后，当您调用函数时，将new 参数指定为字符串：

> df <- grouptest(df, GroupId, IndId, PropertyType, "ValidGroup")   
> df
  GroupId IndId IndGroupProperty PropertyType ValidGroupC ValidGroup
1       1     1                1            1           1          1
2       1     1                2            2           1          1
3       1     2                1            1           1          1
4       1     2                2            2           1          1
5       2     3                3            2           0          0
6       2     4                3            2           0          0
7       2     4                4            1           0          0
8       3     5                5            2           0          0
9       3     5                6            2           0          0

Answer 2

如果对象总是data.frame，为什么不干脆新建一个？

tmp <- data.frame(object, new=tmp[,4])
names(tmp)[4] <- as.character(match.call()$new)
return(tmp)

编辑：将代码更改为接受 name 而不是 character 作为新参数。不过，我仍然认为这不是一个好主意。根据@hadley 在this 线程中的推理，您至少应该有一个可选参数来将第二行切换为names(tmp)[4] <- new。

Answer 3

我怀疑您正在寻找assign 函数：

assign(deparse(substitute(new)), tmp[,4])

显然我误解了这个问题。这是另一种方法。除了使用cbind，您只需将新列添加到现有对象即可。

object[, deparse(substitute(new))] <- tmp[,4]
object