有没有一种方法可以用dplyr来左右填充而不是只向下/向上填充?

时间:2019-04-14 01:13:50

标签: r dplyr

我正在寻找一种使用dplyr将NA填充到右侧(而不是向下/向上)的方法。换句话说,我想将d转换为d2,而不必在mutate调用中显式引用任何列。我的真实数据框有几十个字段,其中NA的交错块跨越可变的列数。我很好奇,不管它出现在哪个字段中,是否有一种简短的方法可以全局地继承左侧的第一个非NA值。

d<-data.frame(c1=c("a",1:4), c2=c(NA,2,NA,4,5), c3=c(NA,3,4,NA,6))
d2<-data.frame(c1=c("a",1:4), c2=c("a",2,2,4,5), c3=c("a",3,4,4,6))
d
d2

2 个答案:

答案 0 :(得分:5)

我们可以将gather转换为“长”格式,将fill按行号分组,然后将spread恢复为“宽”格式

library(tidyverse)
rownames_to_column(d, 'rn') %>% 
    gather(key, val, -rn) %>%
    group_by(rn) %>% 
    fill(val) %>% 
    spread(key, val) %>%
    ungroup %>%
    select(-rn)
# A tibble: 5 x 3
#  c1    c2    c3   
#  <chr> <chr> <chr>
#1 a     a     a    
#2 1     2     3    
#3 2     2     4    
#4 3     4     4    
#5 4     5     6 

或不进行重塑的另一种方法是使用na.locf

进行行填充
library(zoo)
d %>% 
    mutate(c1 = as.character(c1)) %>%
    pmap_dfr(., ~ na.locf(c(...)) %>%
                      as.list %>%
                      as_tibble)

此外,如果我们使用na.locf,它会按列运行,因此数据可以转置并直接应用na.locf

d[] <- t(na.locf(t(d)))
d
#  c1 c2 c3
#1  a  a  a
#2  1  2  3
#3  2  2  4
#4  3  4  4
#5  4  5  6

就像@ G.Grothendieck在评论中提到的那样,为了照顾在行首的NA元素,请使用na.locf0而不是na.locf

答案 1 :(得分:2)

我们可以使用NumericEncoder逐行应用zoo::na.locf

apply