R/create_cohorts.R

In schifferl/LagSelectionBias:

create_cohorts <- function(simulation_years, simulation_races, simulation_sexes,
                           simulation_seed, cohort_size, education_covariance) {
  set.seed(simulation_seed)

  early_life_ses <- rnorm(cohort_size)
  educational_luck <- rnorm(cohort_size)
  educational_ses <- mapply(weighted_ses, early_life_ses, educational_luck,
                            education_covariance)

  list_length <-
    length(simulation_years) *
    length(simulation_races) *
    length(simulation_sexes)

  simulation_data <- vector("list", list_length)

  i <- 1

  for (x in simulation_years) {
    for (y in simulation_races) {
      for (z in simulation_sexes) {
        simulation_data[[i]] <- data.frame(date_of_birth = x, race = y, sex = z,
                                           el_ses, educational_luck,
                                           education = "Missing/Unknown",
                                           date_of_death = integer())

        i <- i + 1
      }
    }
  }

  simulation_data %<>%
    bind_rows()
}