Combine rows and remove NA's

时光怂恿深爱的人放手 提交于 2019-12-13 03:37:29

问题


I have a data frame like this:

idx  type  val1 val2 val3 val4 val5 val6
1    a     0.2   NA   NA   NA   NA   NA
2    a     0.3   NA   NA   NA   NA   NA 
3    a     0.2   NA   NA   NA   NA   NA
4    a     NA    0.3  NA   NA   NA   NA 
5    a     NA    0.5  NA   NA   NA   NA
6    a     NA    0.2  NA   NA   NA   NA
7    a     NA    NA   0.2  NA   NA   NA
8    a     NA    NA   0.5  NA   NA   NA
9    a     NA    NA   0.4  NA   NA   NA
10   a     NA    NA   NA   0.4  NA   NA
11   a     NA    NA   NA   0.6  NA   NA
12   a     NA    NA   NA   0.6  NA   NA
.
. 
.
34   b     NA    NA   NA   NA   NA   0.6
35   b     NA    NA   NA   NA   NA   0.4
36   b     NA    NA   NA   NA   NA   0.3

I want to combine the rows and remove the NA's. So this is what I want to achieve:

idx  type  val1 val2 val3 val4 val5 val6
1    a     0.2  0.3  0.2  0.4  0.3  0.2
2    a     0.3  0.5  0.5  0.6  0.4  0.5
3    a     0.2  0.2  0.4  0.6  0.5  0.6
4    b     0.4  0.2  0.2  0.5  0.4  0.6
5    b     0.3  0.5  0.6  0.3  0.6  0.4
6    b     0.3  0.4  0.3  0.6  0.5  0.3

回答1:


You might just want to apply a function to each of the columns of interest and then cbind it to the category column filtered:

test.df = data.frame(cat = rep(c('a','b'), each=3),x1=c(1,NA,NA,5.5,NA,NA),
                     x2=c(NA,2,NA,NA,4.5,NA),
                     x3=c(NA,NA,3.,NA,NA,3.5))

collapse_column <- function(data, col){
  data[!is.na(data[,col]),col]
}

main_vals = sapply(2:4,collapse_column, data=test.df)

cat_vals = test.df[!is.na(test.df[,2]),'cat']

new_df = as.data.frame(cbind(cat_vals, main_vals))
names(new_df) = names(test.df)


来源:https://stackoverflow.com/questions/47340875/combine-rows-and-remove-nas

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!