R: Заполните все NA в наборе столбцов значениями выше - PullRequest
Новая
       16

R: Заполните все NA в наборе столбцов значениями выше

0 голосов
/ 14 февраля 2019

У меня есть фрейм данных и векторы имен столбцов "a" и столбцов "b": 

x <- data.frame(a1 = c(1, NA, rep(1, 3), NA),
                a2 = c(2, NA, rep(2, 3), NA),
                a3 = c(3, NA, rep(3, 3), NA),
                b1 = c(10, 10, NA, rep(10, 2), NA),
                b2 = c(20, 20, NA, rep(20, 2), NA),
                b3 = c(30, 30, NA, rep(30, 2), NA),
                c = c(2, 3, 5, NA, 9, 8))
avars <- names(x)[1:3]
bvars <- names(x)[4:6]

Есть ли элегантный способ - использовать векторы имен динамических переменных 'avars' и 'bvars'- заполнить все NA в avars и bvars значениями над ними.

Я понимаю, я мог бы использовать цикл вроде этого: 

library(tidyr)
for(i in c(avars, bvars)) x <- x %>% fill(!!i)
x

Но, возможно, есть более элегантныйрешение?Спасибо!

Ответы [ 2 ]

0 голосов
/ 14 февраля 2019

Вы можете использовать tidyr::fill() вместе с grep, чтобы убедиться, что мы заполняем только avars и bvars: 

library(tidyverse)

x %>% fill(grep("^[ab]", names(.)))

  a1 a2 a3 b1 b2 b3  c
1  1  2  3 10 20 30  2
2  1  2  3 10 20 30  3
3  1  2  3 10 20 30  5
4  1  2  3 10 20 30 NA
5  1  2  3 10 20 30  9
6  1  2  3 10 20 30  8

Выражение RegEx ^[ab] утверждает, что имя столбца имеетначать с a или b

или за ваш комментарий, используя avars и bvars: 

x %>% fill(grep(paste0(c(avars,bvars), collapse = "|"), names(x)))

Что все еще лучше, чем цикл forрешение, потому что оно векторизовано.

0 голосов
/ 14 февраля 2019

Используйте na.locf из пакета зоопарка 

> library(zoo)
> na.locf(x)
  a1 a2 a3 b1 b2 b3
1  1  2  3 10 20 30
2  1  2  3 10 20 30
3  1  2  3 10 20 30
4  1  2  3 10 20 30
5  1  2  3 10 20 30
6  1  2  3 10 20 30
...