r - cumsum is.na 与 rle 忽略 consectives NA's
问题描述
简单的问题。假设我有以下数据:
library(tidyverse)
df <- data.frame(group = c(1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2),
variable = c(NA, "a", NA, "b", "c", NA, NA, NA, NA, "a", NA, "c", NA, NA, "d", NA, NA, "a"))
df
group variable
1 1 <NA>
2 1 a
3 1 <NA>
4 1 b
5 1 c
6 1 <NA>
7 1 <NA>
8 1 <NA>
9 1 <NA>
10 1 a
11 1 <NA>
12 1 c
13 1 <NA>
14 1 <NA>
15 1 d
16 2 <NA>
17 2 <NA>
18 2 a
我只想使用cumsum(is.na(variable)
但忽略连续丢失的变量来计算丢失的变量,所以我想要的输出看起来像:
group variable newvariable
1 1 <NA> 1
2 1 a 1
3 1 <NA> 2
4 1 b 2
5 1 c 2
6 1 <NA> 3
7 1 <NA> 3
8 1 <NA> 3
9 1 <NA> 3
10 1 a 3
11 1 <NA> 4
12 1 c 4
13 1 <NA> 5
14 1 <NA> 5
15 1 d 5
16 2 <NA> 1
17 2 <NA> 1
18 2 a 1
我想我需要合并rle
到我的代码中:
df %>%
group_by(group, na_group = {na_group = rle(variable); rep(seq_along(na_group$lengths), na_group$lengths)}) %>%
mutate(newvariable = cumsum((is.na(variable)))) #?
也许map
超过组可以工作。请问有什么建议吗?
解决方案
df %>%
group_by(group) %>%
mutate(new = with(rle(is.na(variable)), rep(cumsum(values), lengths))) %>%
ungroup()