Group_byid并計算連續的NA，然后在遇到新的NA序列時重新開始計數

我有一個這樣的資料框：

df <- data_frame(id = c(rep('A', 10), rep('B', 10)),
                 value = c(1:3, rep(NA, 2), 1:2, rep(NA, 3), 1, rep(NA, 4), 1:3, rep(NA, 2)))

我需要計算value列中連續 NA 的數量。計數需要按 ID 分組，并且每次遇到新的 NA 或新的 NA 序列時都需要從 1 重新開始。預期的輸出應如下所示：

df$expected_output <- c(rep(NA, 3), 1:2, rep(NA, 2), 1:3, NA, 1:4, rep(NA, 3), 1:2)

如果有人能給我一個 dplyr 解決方案，那也很棒:)

我已經嘗試了一些東西，但沒有任何東西給出任何有意義的結果。提前致謝^！

uj5u.com熱心網友回復：

使用dplyr和的解決方案data.table。

library(dplyr)
library(data.table)

df2 <- df %>%
  group_by(id) %>%
  mutate(info = rleid(value)) %>%
  group_by(id, info) %>%
  mutate(expected_output = row_number()) %>%
  ungroup() %>%
  mutate(expected_output = ifelse(!is.na(value), NA, expected_output)) %>%
  select(-info)
df2
# # A tibble: 20 x 3  
#     id    value expected_output
#     <chr> <dbl>           <int>
#  1 A         1              NA
#  2 A         2              NA
#  3 A         3              NA
#  4 A        NA               1
#  5 A        NA               2
#  6 A         1              NA
#  7 A         2              NA
#  8 A        NA               1
#  9 A        NA               2
# 10 A        NA               3
# 11 B         1              NA
# 12 B        NA               1
# 13 B        NA               2
# 14 B        NA               3
# 15 B        NA               4
# 16 B         1              NA
# 17 B         2              NA
# 18 B         3              NA
# 19 B        NA               1
# 20 B        NA               2

uj5u.com熱心網友回復：

這是使用的解決方案rle：

x <- rle(is.na(df$value))
df$new[is.na(df$value)] <- sequence(x$lengths[x$values])

# A tibble: 20 x 3
   id    value   new
   <chr> <dbl> <int>
 1 A         1    NA
 2 A         2    NA
 3 A         3    NA
 4 A        NA     1
 5 A        NA     2
 6 A         1    NA
 7 A         2    NA
 8 A        NA     1
 9 A        NA     2
10 A        NA     3
11 B         1    NA
12 B        NA     1
13 B        NA     2
14 B        NA     3
15 B        NA     4
16 B         1    NA
17 B         2    NA
18 B         3    NA
19 B        NA     1
20 B        NA     2

uj5u.com熱心網友回復：

我們可以使用rle來獲取是或不是 的組的長度na，并用于purrr::map2應用seq它們是否是NA并獲得不斷增長的計數或僅使用 填充NA值rep。

library(tidyverse)

count_na <- function(x) {
  r <- rle(is.na(x))
  consec <- map2(r$lengths, r$values, ~ if (.y) seq(.x) else rep(NA, .x))
  unlist(consec)
}

df %>%
  mutate(expected_output = count_na(value))
#> # A tibble: 20 × 3
#>    id    value expected_output
#>    <chr> <dbl>           <int>
#>  1 A         1              NA
#>  2 A         2              NA
#>  3 A         3              NA
#>  4 A        NA               1
#>  5 A        NA               2
#>  6 A         1              NA
#>  7 A         2              NA
#>  8 A        NA               1
#>  9 A        NA               2
#> 10 A        NA               3
#> 11 B         1              NA
#> 12 B        NA               1
#> 13 B        NA               2
#> 14 B        NA               3
#> 15 B        NA               4
#> 16 B         1              NA
#> 17 B         2              NA
#> 18 B         3              NA
#> 19 B        NA               1
#> 20 B        NA               2

標籤：r dplyr

上一篇：用總結停止警告

下一篇：R中方括號的作用