时间戳

首页 > 解决方案 > data.table 将所有单元格值右移每行内的 na 数 [R]

r - data.table 将所有单元格值右移每行内的 na 数 [R]

问题描述

如何将数据表中的单元格向右移动 R 中每行中的 NA 数？

示例数据：

data <- data.table(c1=c("a","e","h","j"),
                   c2=c("b","f","i",NA),
                   c3=c("c","g",NA,NA),
                   c4=c("d",NA,NA,NA), stringsAsFactors = F)
  c1   c2   c3   c4
1  a    b    c    d
2  e    f    g <NA>
3  h    i <NA> <NA>
4  j <NA> <NA> <NA>

示例中的所需数据：

data.desired <- data.table(
                  c1=c("a",NA,NA,NA),
                   c2=c("b","e",NA,NA),
                   c3=c("c","f","h",NA),
                   c4=c("d","g","i","j"), stringsAsFactors = F)
    c1   c2   c3 c4
1    a    b    c  d
2 <NA>    e    f  g
3 <NA> <NA>    h  i
4 <NA> <NA> <NA>  j

标签： rdataframedata.tableshift

解决方案

NA这是使用矩阵索引和逐行值计数器的一次尝试：

#convert back to a data.frame to take advantage of matrix indexing
setDF(data)   

arr <- which(!is.na(data), arr.ind=TRUE)
arr[,"col"] <- arr[,"col"] + rowSums(is.na(data))[arr[,"row"]]
out <- data
out[] <- NA
out[arr] <- data[!is.na(data)]

out
#    c1   c2   c3 c4
#1    a    b    c  d
#2 <NA>    e    f  g
#3 <NA> <NA>    h  i
#4 <NA> <NA> <NA>  j

#convert to data.table if necessary
setDT(out)

这个选项非常快，从一个简短的测试开始，4 列 / 200 万行大约需要 3-4 秒。

推荐阅读