将NA移至底部

时间:2016-07-22 12:39:39

标签: r data.table

我正在寻找一种简单的方法,将至少有一个NA的所有行移动到dataframe / datatable的底部。例如:

> df <- data.table(aaa=c(1,2,3,4,NA,6,7),
                   bbb=c(1,9,5,NA,3,NA,9),
                   ccc=c(NA,3,NA,4,8,NA,2)
                   )
> df
   aaa bbb ccc
1:   1   1  NA
2:   2   9   3
3:   3   5  NA
4:   4  NA   4
5:  NA   3   8
6:   6  NA  NA
7:   7   9   2

会变成这样:

> df2 <- moveNAtoBottom(df)
> df2
   aaa bbb ccc
1:   2   9   3
2:   7   9   2
3:   1   1  NA
4:   3   5  NA
5:   4  NA   4
6:  NA   3   8
7:   6  NA  NA

即。顶部没有NA的所有列。

注意:

  • 现在,我真的不在乎是否对行应用了任何排序,或者如果订单被保留或反转......只要表现在有两个部分,顶部没有NAs ,底部有NAs
  • 这可以在数据帧或数据表上
  • 依赖是好的

更好的是,如果我能够指定必须应用搜索NA的列。例如,仍然使用前面的例子:

> df3 <- moveNAtoBottom(df, applyTo = c("bbb", "ccc"))
> df3
   aaa bbb ccc
1:   2   9   3
2:  NA   3   8
3:   7   9   2
4:   1   1  NA
5:   3   5  NA
6:   4  NA   4
7:   6  NA  NA

由于

4 个答案:

答案 0 :(得分:5)

你可以这样做:

library(dplyr)
df %>% 
  arrange(rowSums(is.na(.)))

给出了:

#  aaa bbb ccc
#1   2   9   3
#2   7   9   2
#3   1   1  NA
#4   3   5  NA
#5   4  NA   4
#6  NA   3   8
#7   6  NA  NA

或者您是否要指定列名称:

df %>% 
  arrange(rowSums(is.na(select_(., 'bbb:ccc'))))

给出了:

#  aaa bbb ccc
#1   2   9   3
#2  NA   3   8
#3   7   9   2
#4   1   1  NA
#5   3   5  NA
#6   4  NA   4
#7   6  NA  NA

答案 1 :(得分:5)

当OP使用data.table时,data.table的一个选项会循环遍历lapply的列,检查它是否为NA(is.na),明确获取行与Reduce求和,并将其用于order'df'的行。

df[order(Reduce(`+`, lapply(df, is.na)))]
#    aaa bbb ccc
#1:   2   9   3
#2:   7   9   2
#3:   1   1  NA
#4:   3   5  NA
#5:   4  NA   4
#6:  NA   3   8
#7:   6  NA  NA

这是一个类似的方法:

df[, anyna := FALSE]
for (k in names(df)) df[!(anyna), anyna := is.na(get(k))]
df[order(anyna)][, anyna := NULL][]

答案 2 :(得分:2)

一个简单的基础R解决方案可能是:

> df <- data.frame(aaa=c(1,2,3,4,NA,6,7),
+                  bbb=c(1,9,5,NA,3,NA,9),
+                  ccc=c(NA,3,NA,4,8,NA,2))
> ok <- complete.cases(df)
> rbind(df[ok,], df[!ok,])
  aaa bbb ccc
2   2   9   3
7   7   9   2
1   1   1  NA
3   3   5  NA
4   4  NA   4
5  NA   3   8
6   6  NA  NA

并且只选择一些列:

> ok <- complete.cases(df[, c("bbb","ccc")])
> rbind(df[ok,], df[!ok,])
  aaa bbb ccc
2   2   9   3
5  NA   3   8
7   7   9   2
1   1   1  NA
3   3   5  NA
4   4  NA   4
6   6  NA  NA

答案 3 :(得分:1)

尝试:

na=apply(df,1,function(x)sum(is.na(x)))

df[order(na),]

或功能

moveNAtoBottom<-function(df, applyTo=NULL){
  df<-as.data.frame(df)
  if (is.null(applyTo)) applyTo=colnames(df)
  na=rowSums(is.na(df[,applyTo,drop=F]))
  data.table(df[order(na),])
}

moveNAtoBottom(df)
moveNAtoBottom(df,c("ccc", "aaa"))

   aaa bbb ccc
1:   2   9   3
2:   4  NA   4
3:   7   9   2
4:   1   1  NA
5:   3   5  NA
6:  NA   3   8
7:   6  NA  NA