0

我正在尝试将函数应用于数据框的每一行。该函数使用字符和数字的组合作为参数。

这不是聚合练习，而是基于使用 char 和 num 变量的一系列条件在现有 data.frame 中定义新字段。所有变量都在 data.frame 中。

我知道该apply函数将数据框行转换为向量，在本例中为 char 向量：我想避免在函数中重新转换数据框变量。

有几篇关于如何使用apply以及plyr跨 data.frame 行运行函数的包的帖子，但是 - 从我的新手的角度来看 - 这些示例不会逐行混合 char 和 num 参数。谢谢。

condsIdx <- function(IDa,IDb,clss){
  if (clss == 'a'){
    y = IDa
  } else if (clss == 'b' && IDb > 14) {
    y = IDb
  } else {
    y = -1
  }
  y
}

df <- data.frame( IDa=c(1,2,3,4,5,6),
                  IDb=c(11,12,13,14,15,16),
                  clss=c('a','a','a','b','b','c'))


df$res <- -1
for ( i in 1:nrow(df) ){
  df$res[i] <- condsIdx(df$IDa[i],df$IDb[i],df$clss[i])
}
4

2 回答 2

3

只需使用mapply或plyr等效项mlply：

mapply(condsIdx,IDa = df$IDa,IDb = df$IDb,clss = df$clss)
[1]  1  2  3 -1 15 -1
于 2013-03-11T21:09:38.680 回答
3

您可以使用ifelse此处获得矢量化解决方案：

transform(df, res =  ifelse (clss == 'a', 
                             IDa,
                             ifelse(clss == 'b' & IDb > 14,IDb,  -1)))


  IDa IDb clss res
1   1  11    a   1
2   2  12    a   2
3   3  13    a   3
4   4  14    b  -1
5   5  15    b  15
6   6  16    c  -1

编辑一个错字错误 && 没有矢量化所以我用 & 替换它。（感谢@joran）

于 2013-03-11T21:11:07.450 回答