Следующее «конкретное» наблюдение переносится в обратном направлении (NOCB) - PullRequest
Новая
       52

Следующее «конкретное» наблюдение переносится в обратном направлении (NOCB)

0 голосов
/ 07 мая 2018

Это мой фрейм данных: 

library(zoo)
library(dplyr)

df <- data.frame(
  id = rep(1:4, each = 4), 
  status = c(
    NA, "a", "c", "a", 
    NA, "c", "c", "c",
    NA, NA, "a", "c",
    NA, NA, "c", "c"),
  otherVar = letters[1:16],
  stringsAsFactors = FALSE)

Для переменной status я хочу, чтобы следующее наблюдение было перенесено назад в пределах группы (id). 

df %>% group_by(id) %>% na.locf(fromLast = TRUE) %>% ungroup

Тем не менее, я хочу, чтобы только мои "с" переносились в обратном направлении, а не "а".

Из состояния переменной:

NA "a" "c" "a" NA "c" "c" "c" NA NA "a" "c" NA NA "c" "c"

Я хочу получить:

NA "a" "c" "a" "c" "c" "c" "c" NA NA "a" "c" "c" "c" "c" "c"

Соответственно: 

data.frame(
  id = rep(1:4, each = 4), 
  status = c(
    NA, "a", "c", "a", 
    "c", "c", "c", "c",
    NA, NA, "a", "c",
    "c", "c", "c", "c"),
  otherVar = letters[1:16],
  stringsAsFactors = FALSE)

Есть ли способ сделать это?

Ответы [ 2 ]

0 голосов
/ 07 мая 2018

После применения na.locf0 проверьте каждую позицию, которая была NA, и, если она сейчас a, верните ее на NA. Если вы хотите перезаписать status, тогда замените вторую строку status2= на status = if_else(is.na(status) & status2 == "a", NA_character_, status2), status2 = NULL) %>%

library(dplyr)
library(zoo)

df %>% 
  group_by(id) %>% 
  mutate(status2 = na.locf0(status, fromLast = TRUE),
         status2 = if_else(is.na(status) & status2 == "a", NA_character_, status2)) %>%
  ungroup

дает: 

# A tibble: 16 x 4
      id status otherVar status2
   <int> <chr>  <chr>    <chr>  
 1     1 <NA>   a        <NA>   
 2     1 a      b        a      
 3     1 c      c        c      
 4     1 a      d        a      
 5     2 <NA>   e        c      
 6     2 c      f        c      
 7     2 c      g        c      
 8     2 c      h        c      
 9     3 <NA>   i        <NA>   
10     3 <NA>   j        <NA>   
11     3 a      k        a      
12     3 c      l        c      
13     4 <NA>   m        c      
14     4 <NA>   n        c      
15     4 c      o        c      
16     4 c      p        c
0 голосов
/ 07 мая 2018

Решение, использующее tidyr:fill, должно основываться на создании столбца dummyStatus. fill dummyStatus с использованием .direction = "up". Теперь используйте этот dummyStatus для заполнения NA значений в фактическом столбце status после проверки проверки, что следующее значение должно быть c. 

library(dplyr)
library(tidyr)
df %>% group_by(id) %>%
    mutate(dummyStatus = status) %>%
    fill(dummyStatus, .direction = "up" ) %>%
    mutate(status = ifelse(is.na(status) & lead(dummyStatus)=="c","c",status)) %>%
    select(-dummyStatus) %>% as.data.frame()

  #    id status otherVar
  # 1   1   <NA>        a
  # 2   1      a        b
  # 3   1      c        c
  # 4   1      a        d
  # 5   2      c        e
  # 6   2      c        f
  # 7   2      c        g
  # 8   2      c        h
  # 9   3   <NA>        i
  # 10  3   <NA>        j
  # 11  3      a        k
  # 12  3      c        l
  # 13  4      c        m
  # 14  4      c        n
  # 15  4      c        o
  # 16  4      c        p

Данные: 

df <- data.frame(
  id = rep(1:4, each = 4), 
  status = c(
    NA, "a", "c", "a", 
    NA, "c", "c", "c",
    NA, NA, "a", "c",
    NA, NA, "c", "c"),
  otherVar = letters[1:16],
  stringsAsFactors = FALSE)
...