如何在 data.frame 中动态创建新列？

Question

类似问题

How to name variables on the fly?

基本上，我想随时随地在 data.frame 中创建一个新列。可能是不好的做法，但对于我正在做的事情来说，这是一个很好的解决方案。

现在我已经试过了：

test <- iris
create.new_var <- function(x) {
  assign(paste("test$", x, sep=""), test$Petal.Width)
  return(test)
}

test <- create.new.var('cheese') 

函数运行时不会中断。但是 data.frame 测试不包含标题为“奶酪”的新列和 iris$Petal.Width 的值，正如人们想象的那样。

Answer 1

不建议使用assign 进行此类操作。但是，如果您需要尝试使用assign，一个可能的选择是

 create.new_var <- function(x){
  assign('test', `[[<-`(test, x, value=test$Petal.Width), envir=.GlobalEnv)
 }
 test <- create.new_var('cheese')
 head(test,3)
 #  Sepal.Length Sepal.Width Petal.Length Petal.Width Species cheese
 #1          5.1         3.5          1.4         0.2  setosa    0.2
 #2          4.9         3.0          1.4         0.2  setosa    0.2
 #3          4.7         3.2          1.3         0.2  setosa    0.2

您还可以在函数中添加数据和替换列参数

create.new_var <- function(dat, x, x1){
  str1 <- deparse(substitute(dat))
  assign(str1, `[[<-`(dat, x, value=dat[,x1]), envir=.GlobalEnv)
}

test <- create.new_var(test, 'cheese', 'Petal.Width')

这是一个不使用assign或paste的选项（一些被删除的帖子也使用了类似的方法）。

 create.new_var2 <- function(dat, x, x1){ 
                       dat[,x] <- dat[,x1]
                       dat}

 create.new_var2(test, 'cheese', 'Petal.Width')