data.table

Question

के साथ एक सरल पाश मैं data.table का उपयोग कर एक सरल पाश करना चाहते हैं। मैं 20 दिचोतोमोउस (0,1) चर (var_20 को VAR_1 से) है और मैं इस के लिए एक पाश करना चाहते हैं:

dat[var_1==1, newvar:=1] 
dat[var_2==1, newvar:=2] 
dat[var_3==1, newvar:=3] 
... 
dat[var_20==1, newvar:=21] 

मेरा मुख्य समस्या मैं कैसे निर्दिष्ट मैं (यानी VAR_1 = नहीं जानता है = 1, var_2 == 2 ...) एक लूप का उपयोग कर। एक छोटी उदाहरण नीचे:

var_1 <- c(1, rep(0,9)) 
var_2 <- c(0,1, rep(0,8)) 
var_3 <- c(0,0,1, rep(0,7)) 
dat <- data.table(var_1, var_2, var_3) 

dat[var_1==1, newvar:=1] 
dat[var_2==1, newvar:=2] 
dat[var_3==1, newvar:=3] 

कैसे एक पाश के साथ ऐसा करने के बारे में कोई भी विचार? धन्यवाद!

Answer 1

डेटाटेबल क्लास का लाभ लेने के लिए कुंजी सेट करना बेहतर है।

dat[ ,newvar:= NA_integer_] 
for(i in ncol(dat)) { 
setkeyv(dat, names(dat)[i]) 
dat[J(1), newvar:=i] 
} 

Answer 2

कुछ इस तरह काम करेगा।

nams <- names(dat) 
for(n in seq_along(nams)){ 
    nam <- nams[n] 
    char <- sprintf('%s==1',nam) 
    dat[eval(parse(text=char)), newvar := n] 
} 
dat 
var_1 var_2 var_3 newvar 
1:  1  0  0  1 
2:  0  1  0  2 
3:  0  0  1  3 
4:  0  0  0  NA 
5:  0  0  0  NA 
6:  0  0  0  NA 
7:  0  0  0  NA 
8:  0  0  0  NA 
9:  0  0  0  NA 
10: 0  0  0  NA