R/tble_strp.R

In csmoxford/prettyTables: Pretty tables

.tbl_strp = function(tble, strata, all_strata, data, var, var.order, type, nxt_row, rnd, messages, zeros){

  data = data[!is.nan(data[[var]]) & !(data[[var]] == "NaN"),]
  if(dim(data)[1] > 0) {
    if(sum(is.na(data[[var]])) > 0) {
      data[is.na(data[[var]]), var] = "Missing"
    }
    data[data[[var]] == "missing", var] = "Missing"
    data[data[[var]] == "", var] = "Missing"
  }

  # Get variable ordering
  if(var %in% names(var.order)){
    st=var.order[[var]]
    if(!all(unique(sort(data[[var]])) %in% sort(var.order[[var]])) & messages){
      message("Not all values for ", var, " match values provided. Provided values are: ", paste(sort(var.order[[var]]),collapse = ", "), " Values found are: ", paste(sort(unique(data[[var]])),collapse = ", "))
    }
  } else {
    st = sort(unique(data[[var]]))
    if("Other" %in% st | "other" %in% st) {
      which.other = ifelse("Other" %in% st,"Other","other")
      st=c(st[-which(st == which.other)],which.other)
    }
    if("Missing" %in% st) {
      st=c(st[-which(st == "Missing")],"Missing")
    }
  }

  for(i in st) {
    tble[nxt_row, 2] = i
    for(j in all_strata) {
      if(dim(data)[1]>0) {
        if(j=="Overall") {
          data_sub = data[, var]
        } else {
          data_sub = data[data[strata] == j, var]
        }

        ####################################
        # section to change for other outputs
        num = sum(data_sub == i, na.rm=TRUE)
        den = length(data_sub)
        if(num == 0 & !zeros){
          tble[nxt_row, j] = ""
        }
        if(den > 0) {
          tble[nxt_row, j] = paste0(num, "/", den," (", roundWZero(num * 100 / den, rnd), "%)")
        } else {
          tble[nxt_row, j] = "0/0 (0%)"
        }

        #####################################
      } else {
        tble[nxt_row, j] = "0/0 (0%)"
      }
    }
    nxt_row = nxt_row + 1
  }

  return(list(tble=tble,nxt_row=nxt_row))

}