使用用户定义的函数在 R 中创建一个新列

Question

我正在尝试对列进行模式匹配，并使用用户定义的函数使用匹配模式的一部分创建一个新列。

表名是数据，它包含

MatchedColumn
24 abcd
32 gcef
45 lmno

代码是

f4<-function()
{
  data$NewColumn <- gsub('([0-9]{2}) *','\\1',data$MatchedColumn)
}

我将函数调用为 f4()，但它没有执行函数的内部部分。

但是当我单独运行以下行时，它会在匹配后创建一个新列，生成的表如图所示。

data$NewColumn <- gsub('([0-9]{2}) *','\\1',data$MatchedColumn)

数据：

MatchedColumn  NewColumn
 24 abcd         24
 32 gcef         32
 45 lmno         45

我正在尝试使用用户定义的函数，因为我想通过函数传递 NewColumn 和 MatchedColumn 名称，例如 f4(NewColumn,MatchedColumn)

如果有人知道怎么做，请帮忙！

Answer 1

有几种选择：

• 您可以将数据作为参数传递给函数

  f4<-function(data, col1, col2)
  {
     data[,col1] <- sapply(data[col2], function(x)gsub("[^0-9]", "", x))
     return(data)
  }
  
  newdata <- f4(data, 'NewColumn', 'MatchedColumn')
  str(newdata)
  

• 您可以使用<<- 进行全局分配

  f4<-function(col1, col2)
  {
    data[col1] <<- sapply(data[col2], function(x)gsub("[^0-9]", "", x))
  }
  
  f4('NewColumn', 'MatchedColumn')
  

我肯定更喜欢第一个选项