Làm thế nào để hợp nhất tất cả các yếu tố của một danh sách với một bảng số liệu về tình trạng

Question

tôi có các dữ liệu sau:

library(data.table) 

dt1 <- data.table(var1 = c("wk1","wk1","wk2"), 
        var2 = c(1,2,3)) 

dt2 <- data.table(var3 = c("a","b","c"), 
        var2 = c(1,2,3)) 

lista <- list(dt1,dt2) 

dt_main <- data.table(var1 = c("wk1","wk2"), 
         var4 = c(100,200)) 

Tôi muốn merge tất cả các yếu tố của lista có chứa biến var1 với dt_maindata.table, vì vậy trong cuối tôi muốn lista trông như thế này:

dt1 <- data.table(var1 = c("wk1","wk1","wk2"), 
        var2 = c(1,2,3), 
        var4 = c(100,100,200)) 

dt2 <- data.table(var3 = c("a","b","c"), 
        var2 = c(1,2,3)) 

lista <- list(dt1,dt2) 

tôi đã cố gắng

mapply(function(X,Y){ 
    if("var1"%in%names(X)){ 
    X <- merge(X,Y,by="var1") 
    } 
},X=lista,Y=dt_main) 

nhưng không hoạt động. Bất kỳ giúp đỡ?

Answer 1

Bạn có thể sử dụng một lapply và hợp nhất bên trong hàm:

lapply(lista, function(x) if (!is.null(x$var1)) { 

    #the function checks if there is a var1 column 
    #and if there is, it gets merged to the x data.table 
    return(merge(dt_main, x, by = 'var1', all.x = TRUE)) 

} else { 

    #otherwise it just returns the data.table 
    return(x) 

}) 

# [[1]] 
# var1 var4 var2 
# 1: wk1 100 1 
# 2: wk1 100 2 
# 3: wk2 200 3 
# 
# [[2]] 
# var3 var2 
# 1: a 1 
# 2: b 2 
# 3: c 3 

Answer 2

Một cách hơi khác để làm điều này:

lapply(lista, function(x) if ('var1' %in% names(x)) 
    x[dt_main, on = 'var1', var4 := var4][] 
    else x 
) 

mang đến cho:

[[1]] 
    var1 var2 var4 
1: wk1 1 100 
2: wk1 2 100 
3: wk2 3 200 

[[2]] 
    var3 var2 
1: a 1 
2: b 2 
3: c 3