Обновление значений с использованием lapply в data.table условно для nrow

Question

Я хочу обновить значения столбца в таблице data.table, привязанные к двум другим столбцам в одной таблице.

Этот код здесь работает некорректно, но я хотел обновить значения в своем DT вместо того, чтобы возвращать список DT.

воспроизводимый пример:

library(data.table) 
library(gtools) 

# Create Data Table 
    DT <- as.data.table(combinations(3,2,1:3,repeats=TRUE)) 
    DT[, V3 := 9999 ] 

DT 
> V1 V2 V3 
>1: 1 1 9999 
>2: 1 2 9999 
>3: 1 3 9999 
>4: 2 2 9999 
>5: 2 3 9999 
>6: 3 3 9999 

Мой код:

# create function 

stationary <- function(i) {DT[i, V3 := if (V1==V2) 0 else V1+V2 ]} 

i <- 1:nrow(DT) 

DT <- lapply(i, stationary) # This returns a list of identical data tables 

В результате я хотел:

DT 
> V1 V2 V3 
>1: 1 1 0 
>2: 1 2 3 
>3: 1 3 4 
>4: 2 2 0 
>5: 2 3 5 
>6: 3 3 0 

Answer 1

Я хотел бы сделать это:

DT[, V3 := (V1 + V2) * (V1 != V2)] 
# V1 V2 V3 
#1: 1 1 0 
#2: 1 2 3 
#3: 1 3 4 
#4: 2 2 0 
#5: 2 3 5 
#6: 3 3 0 

Это быстро и просто.

Answer 2

Вы можете использовать цепочки:

library(data.table) 
library(gtools) # combinations() 
DT <- as.data.table(combinations(3,2,1:3,repeats=TRUE)) 
DT[, V3 := 9999 ] 
DT[V1==V2, V3 := 0 
    ][V1!=V2, V3 := as.numeric(V1+V2) 
    ][] 
# V1 V2 V3 
# 1: 1 1 0 
# 2: 1 2 3 
# 3: 1 3 4 
# 4: 2 2 0 
# 5: 2 3 5 
# 6: 3 3 0 

Вы можете избежать as.numeric вызова, если вы вставляете с целыми числами, так и V3 := 9999LV3 := 0L.