R/simulate_surveydata.R

In stefmue/georefum: Processing and Linking of Spatial Data Within the GeorefUm Project at GESIS

# simulate_surveydata.R
#' Simulate survey data
#'@param N Number of respondents that should be simulated
#'@param na Number of missing values in attributes
#'@export

simulate_surveydata <- function(N = 10000, na = 500) {

  # sinulate survey data attributes --------------------------------------------
  return(
    data.frame(

      # id -----------------------------------------------------------------
      id = stringr::str_pad(sample(1:N, N, replace = FALSE), nchar(N),
                            pad = "0"),

      # age ----------------------------------------------------------------
      age = round(rnorm(N, mean = 50, sd = 10)),

      # gender -------------------------------------------------------------
      gender = sample(1:2, 10000, replace = TRUE, prob = c(.5, .5)),

      # att1 ---------------------------------------------------------------
      att1 = {attr <- sample(1:5, N, replace = TRUE,
                             prob = c(.4, .2, .3, .05, .05));
      attr[sample(1:N, na)] <- NA; attr},

      # att2 ---------------------------------------------------------------
      att2 = {attr <- sample(1:5, N, replace = TRUE,
                    prob = c(.2, .2, .2, .3, .1));
      attr[sample(1:N, na)] <- NA; attr},

      # att3 ---------------------------------------------------------------
      att3 = {attr <- sample(1:10, N, replace = TRUE,
                    prob = c(.05, .1, .4, .1, .05, .05, .1, .025, .025,
                             .05));
      attr[sample(1:N, na)] <- NA; attr}
    )
  )
}