thêm giá trị cột dựa trên các cột khác trong khung dữ liệu sử dụng cho và nếu

Question

Tôi có một dataframe như thế này:

 id  adit  diag1 diag2  
     2  3   4230 2234   
     3  5   3345 4456   
     4  6   4567 4467 

Tôi muốn thêm 2 cột khác, dse1 và dse2 sử dụng pseudo-code dưới đây:

if diag1 contains 4230 then dse1 = 1 else dse1 = 0 

if diag2 contains 4567 then dse2 =1 else dse2 = 0 

tôi sử dụng này:

for (i in 1 : nrow(dse)){ 
    for (j in 3: ncol(dse)){ 
    if dse[i,j] %in% ("4320"){dse$dse1 = 1} 
     else{dse$dse1 = 0} 
    if dse[i,j] %in% ("4567"){dse$dse2 = 1} 
     else{dse$dse2 = 0} 
    } 
} 

Nhưng những làm không t làm việc.

Answer 1

Không cần phải sử dụng một vòng lặp, chỉ cần sử dụng ifelse, ví dụ

dse = within(dse, { 
    dse1 = ifelse(diag1 == 4230, 1, 0) 
    dse2 = ifelse(diag2 == 4567, 1, 0) 
}) 

Answer 2

Không sử dụng if/else. Go vector hóa như trong:

dat$dse1 <- as.numeric(dat$diag1 == 4230) 
dat$dse2 <- as.numeric(dat$diag2 == 4567) 

Answer 3

Bạn có thể sử dụng transform:

transform(dse, dse1 = as.numeric(diag1 == 4230), 
       dse2 = as.numeric(diag2 == 4567)) 

Answer 4

Như thế này:

dse$dse1<-0 
dse$dse2<-0 
dse$dse1[dse$diag1==4230]<-1 
dse$dse2[dse$diag2==4567]<-1 

Hãy tự mình có một hướng dẫn R tốt (chẳng hạn như this) và đọc tất cả về vectơ chỉ mục.