một vòng lặp đơn giản với data.table

Question

Tôi muốn thực hiện một vòng lặp đơn giản bằng cách sử dụng data.table. Tôi có 20 nhị phân (0,1) biến (từ VAR_1 để var_20) và tôi muốn làm một vòng cho việc này:

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

vấn đề chính của tôi là tôi không biết làm thế nào xác định i (tức là VAR_1 = = 1, var_2 == 2 ...) sử dụng vòng lặp. Dưới đây là một ví dụ ngắn:

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] 

Bất kỳ ý tưởng nào về cách thực hiện điều này bằng vòng lặp? Cảm ơn!

Answer 1

Để tận dụng lợi thế của lớp dữ liệu, tốt hơn nên đặt khóa.

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

Answer 2

Điều gì đó như thế này sẽ hoạt động.

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