根據數據幀中的條件填寫新列

Question

我想在我的數據框中添加一個新列，以便對於每一行如果LOC == 1那麼V等於V1的給定值;如果LOC==2則V等於V2給出的值。這裏有一個例子：

df <- 
LOC V1 V2 
1 0.5 0.7 
1 0.5 0.7 
2 0.5 0.7 
1 0.6 0.8 

的結果應該是：

df <- 
LOC V1 V2 V 
1 0.5 0.7 0.5 
1 0.5 0.7 0.5 
2 0.5 0.7 0.7 
1 0.6 0.8 0.6 

我需要怎麼做幫助，在R.

Answer 1

如果LOC只包含1或2，那麼這將工作

df$V <- ifelse(df$LOC == 1, df$V1, df$V2) 

Answer 2

使用矩陣索引：

idx <- cbind(seq_len(nrow(dat)), dat$LOC) 
#  row col 
#  [,1] [,2] 
#[1,] 1 1 
#[2,] 2 1 
#[3,] 3 2 
#[4,] 4 1 

dat[-1][idx] 
#[1] 0.5 0.5 0.7 0.6