R/internal_synthesize_sample_categorical.R

In simongrund1/robosynth: Robust Synthetic Data Generation with Data-Augmented Multiple Imputation

# // Sampling for categorical data

sample.categorical <- function(n = nrow(probs), k = ncol(probs), probs) {

  u <- runif(n)

  if (k == 2L) {

    y <- u > probs[, 1]

  } else {

    y <- cp <- numeric(n)

    for (j in 1L:(k-1L)) {
      cp <- cp + probs[, j]
      y[u > cp] <- j
    }

  }
    
  return(y + 1L)

}