Измените на длинный и полный с NA

Question

Мне нужно изменить форму df, дополнить ее пропущенными годами и создать переменное состояние отслеживания изменений.Проблема в том, что некоторые значения отсутствуют, и код, который я написал, тормозит их.

Пример игрушки:

library(data.table)
df <- data.frame(id=c(1,2),phase_1=c(1994,1994),phase_2=c(1996,1996),phase_3=c(1997,NA))
df1 = melt(df, 
         id.vars = "id",
         measure.vars = c("phase_1", "phase_2", "phase_3"),
         variable.name = "status",
         value.name = "year",
         na.rm = FALSE)
df2 <- df1 %>%  complete(id, year = full_seq(year, 1)) %>% 
  fill(status)

Желаемый

  id year   phase change
1  1 1994 phase_1      0
2  1 1995 phase_1      0
3  1 1996 phase_2      1
4  1 1997 phase_3      1
5  2 1994 phase_1      0
6  2 1995 phase_1      0
7  2 1996 phase_2      1
8  2 1997 phase_2      0

Answer 1

С dplyr и tidyr вы также можете сделать:

df %>%
 gather(phase, year, -id, na.rm = TRUE) %>%
 complete(id, year = full_seq(year, 1)) %>%
 fill(phase) %>%
 group_by(id) %>%
 mutate(change = as.numeric(phase != lag(phase, default = first(phase))))

     id  year phase   change
  <dbl> <dbl> <chr>    <dbl>
1     1  1994 phase_1      0
2     1  1995 phase_1      0
3     1  1996 phase_2      1
4     1  1997 phase_3      1
5     2  1994 phase_1      0
6     2  1995 phase_1      0
7     2  1996 phase_2      1
8     2  1997 phase_2      0

Или:

 df %>%
 gather(phase, year, -id, na.rm = TRUE) %>%
 complete(id, year = full_seq(year, 1)) %>%
 fill(phase) %>%
 group_by(id) %>%
 mutate(change = (phase != lag(phase, default = first(phase))) * 1)

Answer 2

Вы можете использовать dplyr и tidyr как:

library(dplyr)
library(tidyr)
df %>%
  gather(phase, year, phase_1:phase_3) %>%
  filter(!is.na(year)) %>%
  complete(id, year = full_seq(year, 1)) %>%
  mutate(phase = ifelse(is.na(phase), lag(phase,1), phase)) %>%
  group_by(id) %>%  
  mutate(change = ifelse(phase == lag(phase, 1) | row_number() == 1, 0, 1)) 
# A tibble: 8 x 4
# Groups:   id [2]
     id  year phase   change
  <dbl> <dbl> <chr>    <dbl>
1     1  1994 phase_1      0
2     1  1995 phase_1      0
3     1  1996 phase_2      1
4     1  1997 phase_3      1
5     2  1994 phase_1      0
6     2  1995 phase_1      0
7     2  1996 phase_2      1
8     2  1997 phase_2      0