Question

说我有一个data.table

prueba <- data.table(aa=1:7,
                     bb=c(1,2,NA,NA,3,1,1),
                     cc=c(1,2,NA,NA,3,1,1),
                   YEAR=c(1,1,1,2,2,2,2))

我想从给定的集合（例如aa，bb和cc）中获取每一列的缺失表。

结果应该是这样的：

    aa   bb   cc
1:   0    2    2    
0:   7    0    0

或其转置形式或带有其他标签。

我尝试过

prueba[,lapply(.SD, function(x) as.list(  table(
       factor(is.na(x), levels=c("0","1"))))),
       .SDcols=c("aa","bb", "cc")]

但是我却得到了这个：

    aa   bb   cc
1:   7    5    5    
0:   7    2    2

我认为这与表删除未使用的级别有关。但是我没有尝试过使用xtabs和各种黑客手段。

我可以得到一些丑陋的东西

sapply(c("aa","bb","cc"), function(x) prueba[,as.list(
      table(is.na(get(x))))])

Answer 1

也许使用Orders orders = new Orders(); for (int i = 1; i < 10; i++) { Items itm = new Items { ItemName = $"Item {i}" }; orders.ItemList.Add(itm); }：

table

这实际上将其像矩阵一样对待。

一些替代方法：

prueba[, table(is.na(.SD), names(.SD)[col(.SD)]), .SDcols=aa:cc]

        aa bb cc
  FALSE  7  5  5
  TRUE   0  2  2

Answer 2

这是一种以R为底的方法：

rbind(tmp <- colSums(is.na(prueba[ , -"YEAR"])), nrow(prueba) - tmp)
#      aa bb cc
# [1,]  0  2  2
# [2,]  7  5  5

Answer 3

这是另一个应该足够笼统的建议。首先，根据需求建立列联表。接下来，将表输出转换为列表，并将所有结果rbindlist一起转换。最后，将NA替换为0个计数。

output <- prueba[, rbindlist(
        lapply(.SD, function(x) as.list(table(is.na(x)))), 
        fill=TRUE, 
        idcol=TRUE), 
    .SDcols=aa:cc]

output[, lapply(.SD, function(x) replace(x, is.na(x), 0L))]

输出：

   .id FALSE TRUE
1:  aa     7    0
2:  bb     5    2
3:  cc     5    2

编辑：添加另一种通用方法：

#build and flatten contingency table
tab <- prueba[, as.list(unlist(lapply(.SD, function(x) table(is.na(x))))),
    .SDcols=aa:cc]

#melt, split original column names and then pivot
dcast(
    melt(tab, measure.vars=names(tab))[, 
        c("V1","Factor") := tstrsplit(variable, split="\\.")],
    Factor ~ V1, 
    function(x) x[1L], 
    fill=0L)

输出：

   Factor aa bb cc
1:  FALSE  7  5  5
2:   TRUE  0  2  2

编辑：添加时间

set.seed(0L)
sz <- 1e6
nc <- 10
DT <- as.data.table(matrix(sample(c(NA_integer_, 1L:10L), sz*nc, TRUE), ncol=nc))
setnames(DT, paste0("C", 1L:nc))
cols <- names(DT)

mtd1 <- function() {
    DT[, table(is.na(.SD), names(.SD)[col(.SD)]), .SDcols=cols]
}

mtd2 <- function() {
    DT[, table(is.na(.SD), rep(names(.SD), each=.N)), .SDcols=cols]
}

mtd3 <- function() {
    melt(DT[, ..cols], measure.vars=cols)[, table(is.na(value), variable)]
}

mtd4 <- function() {
    tab <- DT[, as.list(unlist(lapply(.SD, function(x) table(is.na(x))))),
        .SDcols=cols]

    dcast(melt(tab, measure.vars=names(tab))[, c("V1","Factor") := tstrsplit(variable, split="\\.")],
        Factor ~ V1, function(x) x[1L], fill=0L)
}

mtd5 <- function() {
    output <- DT[, rbindlist(lapply(.SD, function(x) as.list(table(is.na(x)))), fill=TRUE, idcol=TRUE),
        .SDcols=cols]

    output[, lapply(.SD, function(x) replace(x, is.na(x), 0L))]
}

library(microbenchmark)
microbenchmark(mtd1(), mtd2(), mtd3(), mtd4(), mtd5(), times=3L)

时间：

Unit: seconds
   expr      min       lq     mean   median       uq      max neval cld
 mtd1() 5.044369 5.049252 5.086534 5.054135 5.107617 5.161100     3   b
 mtd2() 5.106796 5.110014 5.474269 5.113232 5.658005 6.202778     3   b
 mtd3() 2.395127 2.461463 2.509938 2.527799 2.567344 2.606888     3  a 
 mtd4() 2.138672 2.142300 2.145895 2.145927 2.149506 2.153084     3  a 
 mtd5() 2.113367 2.175346 2.228162 2.237325 2.285560 2.333794     3  a

Answer 4

好的，我找到了一个解决方案，有点令人费解：

prueba[, lapply(.SD, function(x) as.list( table(factor(
is.na(x), levels=c(F,T)))) ), .SDcols=c("aa","bb", "cc")]

应该有一种更简单的方法。

如何使用data.table将表应用于多个列？

4 个答案: