data=data.frame(GROUP=c(1,1,1,2,2,2,3,3,3,4,4,4),
VAR=c('A','B','C','A','B','C','A','B','C','A','B','C'),
SCORE=c(3,6,NA,NA,NA,NA,1,NA,5,5,2,NA),
NEWVAR=c(1,1,1,NA,NA,NA,2,2,2,1,1,1))
score1 = c(2,3,7)
score2 = c(0,5,6)
除了“ NEWVAR”外，我的所有列都有“数据”，我希望像这样创建它：
如果特定“组”的得分为1％in％，则为“ NEWVAR”赋予值“ 1”
如果对于特定的“组”，得分2％in％SCORE和得分1不在SCORE中，则将值'2'赋予'NEWVAR'
由于您已使用in语法标记了此data.table。ifelsedata.table
library(data.table)
setDT(data)[, NEWVAR := ifelse(any(SCORE %in% score1), 1,
ifelse(any(SCORE %in% score2), 2, NA_integer_)), GROUP]
data
# GROUP VAR SCORE NEWVAR
# 1: 1 A 3 1
# 2: 1 B 6 1
# 3: 1 C NA 1
# 4: 2 A NA NA
# 5: 2 B NA NA
# 6: 2 C NA NA
# 7: 3 A 1 2
# 8: 3 B NA 2
# 9: 3 C 5 2
#10: 4 A 5 1
#11: 4 B 2 1
#12: 4 C NA 1
我觉得也有fcase在开发版本的data.table这类似于case_when在dplyr并使得编写这样的嵌套ifelse容易。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句