R/filter_cohort.R

In schifferl/LagSelectionBias:

filter_cohort <- function(simulation_data, current_year, current_race,
                          current_sex) {
  simulation_data %>%
    filter(date_of_birth <= current_year) %>%
    filter(race == current_race) %>%
    filter(sex == current_sex) %>%
    filter(is.na(date_of_death)) %>%
    mutate(age_group = sapply(current_year - date_of_birth, determine_age_group)) %>%
    arrange(date_of_birth, race, sex, theta_education)
}