rbindlist data.tables不同的维度

时间:2016-08-29 04:38:39

标签: r data.table bind lapply

我使用不同的输出多次执行一个函数,如图所示。

require(data.table)
myfunction<-function(x){
DT1<-data.table(a=c(1,2,3),b=c("a","b","c"))
DT2<-data.table(d=c(4,5,6), e=c("d","e","f"))
return(list(DT1=DT1, DT2=DT2))
}
result<-lapply(1:2, myfunction)

我想绑定结果。所需的输出将如我所示。我的真实例子使用了数百个表。

l1<-rbindlist(list(result[[1]]$DT1, result[[2]]$DT1), idcol = TRUE)
l2<-rbindlist(list(result[[1]]$DT2, result[[2]]$DT2), idcol = TRUE)
DESIRED_OUTPUT<-list(l1, l2)

我使用此选项但不起作用: rbindlist data.tables wtih different number of columns

=============================================== =======================

更新

当列表的元素数量不同于2时,@ nicola提出的选项不起作用。对于第一个例子(DT1和DT2)。作为解决方案,我创建了一个变量“l”,用于计算函数列表中元素的数量。

解决方案的新例子。

require(data.table)
myfunction<-function(x){
DT1<-data.table(a=c(1,2,3),b=c("a","b","c"))
DT2<-data.table(d=c(4,5), e=c("d","e"))
DT3<-data.table(f=c(7,8,NA,9), g=c("g","h","i","j"))
return(list(DT1=DT1, DT2=DT2, DT3=DT3))
}
result<-lapply(1:5, myfunction)

l<-unique(sapply(result, length))
apply(matrix(unlist(result,recursive=FALSE),nrow=l),1,rbindlist,idcol=TRUE)

2 个答案:

答案 0 :(得分:1)

应尝试匹配列表组件的名称:

Map(
  function(LL,n) rbindlist(unname(LL[names(l) %in% n]), idcol=TRUE),
  list(unlist(result, recursive=FALSE)),
  unique(names(l))
)

#[[1]]
#   .id a b
#1:   1 1 a
#2:   1 2 b
#3:   1 3 c
#4:   2 1 a
#5:   2 2 b
#6:   2 3 c
#
#[[2]]
#   .id d e
#1:   1 4 d
#2:   1 5 e
#3:   1 6 f
#4:   2 4 d
#5:   2 5 e
#6:   2 6 f

答案 1 :(得分:1)

这是一个选项:

do.call(function(...) Map(function(...) rbind(..., idcol = T), ...), result)
#$DT1
#   .id a b
#1:   1 1 a
#2:   1 2 b
#3:   1 3 c
#4:   2 1 a
#5:   2 2 b
#6:   2 3 c
#
#$DT2
#   .id d e
#1:   1 4 d
#2:   1 5 e
#3:   1 6 f
#4:   2 4 d
#5:   2 5 e
#6:   2 6 f

这是另一个:

lapply(purrr::transpose(result), rbindlist, idcol = T)