基于R中的两个变量的子设置面板数据

Question

library(dplyr)
id <- c(rep(1,4),rep(2,3),rep(3,4))
missing <- c(rep(0,4),rep(0,3),1,0,0,0)
wave <- c(seq(1:4),1,2,3,seq(1:4))
df <- as.data.frame(cbind(id,missing,wave)) 
df



id missing wave
1   1       0    1
2   1       0    2
3   1       0    3
4   1       0    4
5   2       0    1
6   2       0    2
7   2       0    3
8   3       1    1
9   3       0    2
10  3       0    3
11  3       0    4

I am trying to delete cases if they have missing=1 or if they are missing a wave (1:4). For example, ID=3 should be dropped because at wave=1 they have missing=1 and ID=2 should be dropped because they only have values of 1, 2, and 3 in Wave.

I tried to use dplyr's group_by and filter functions but this removes all cases. I want to only end up with cases for ID=1.

df <- df %>% group_by(id) %>% filter(missing==0, wave==1, wave==2, wave==3, wave==4)
df

Answer 1

Try this. We first group_by id , and then create a list column with the sorted unique values of wave for each id . Then we check to make sure this list equals 1:4 . We create a missing_check variable, which is just the max of missing for each id . We filter on both missing_check and wave_check .

df %>%
    group_by(id) %>%
    mutate(wave_list = I(list(sort(unique(wave))))) %>%
    mutate(wave_list_check = all(unlist(wave_list) == 1:4),
           missing_check = max(missing)) %>%
    filter(missing_check == 0, wave_list_check) %>%
    select(id:wave)

     id missing  wave
  <dbl>   <dbl> <dbl>
1     1       0     1
2     1       0     2
3     1       0     3
4     1       0     4