Как суммировать по NA строк по группам в R - PullRequest
Новая
       40

Как суммировать по NA строк по группам в R

2 голосов
/ 31 марта 2020

Допустим, у меня есть следующий фрейм данных: 

mydat <- structure(list(Group = structure(c(1L, 1L, 1L, 2L, 2L, 2L, 2L
), .Label = c("A", "B"), class = "factor"), Day = c(1, 2, 3, 
1, 2, 3, 4), Var1 = c(2, 3, 5, 12, NA, NA, 51), Var2 = c(5, 6, 
2, 0, 40, 50, 3)), class = "data.frame", row.names = c(NA, -7L
))

mydat

  Group Day Var1 Var2
1     A   1    2    5
2     A   2    3    6
3     A   3    5    2
4     B   1   12    0
5     B   2   NA   40
6     B   3   NA   50
7     B   4   51    3

Что я хочу сделать (желательно с dplyr): по группе найти строки, где Var1 равно NA, и над этими суммируйте суммы Var2 и включите эту сумму в следующую строку, в которой Var1 не равно NA. Таким образом: 

mydat_new <- structure(list(Group = structure(c(1L, 1L, 1L, 2L, 2L), .Label = c("A", 
"B"), class = "factor"), Day = c(1, 2, 3, 1, 4), Var1 = c(2, 
3, 5, 12, 51), Var2 = c(5, 6, 2, 0, 93)), class = "data.frame", row.names = c(NA, 
-5L))

mydat_new

  Group Day Var1 Var2
1     A   1    2    5
2     A   2    3    6
3     A   3    5    2
4     B   1   12    0
5     B   4   51   93

Таким образом, в Group B строки со Днями 2 и 3 пропали, и их вклад Var2 был "поглощен" в следующий день, чья Var1 не NA, т. Е. День 4.

1 Ответ

2 голосов
/ 31 марта 2020

Один из вариантов будет: 

library(dplyr)

mydat %>%
  group_by(Group, idx = rev(cumsum(rev(!is.na(Var1))))) %>%
  mutate(Var2 = sum(Var2)) %>%
  ungroup() %>%
  filter(!is.na(Var1)) %>%
  select(-idx)

Вывод: 

# A tibble: 5 x 4
  Group   Day  Var1  Var2
  <fct> <dbl> <dbl> <dbl>
1 A         1     2     5
2 A         2     3     6
3 A         3     5     2
4 B         1    12     0
5 B         4    51    93
...