var_case

Author: astra-cdc

R/var_case.R

@@ -3,20 +3,29 @@
 #' @param newvr   name of the new logical variable to be created
 #' @param vr factor variable
 #' @param cases one or more levels of `vr` that are converted to `TRUE`. All other levels are converted to `FALSE`.
+#' @param retain_na for the observations where `vr` is `NA`, should `newvr` be `NA` as well?
 #'
 #' @return Survey object
 #' @family variables
 #' @export
 #'
 #' @examples
 #' set_survey(namcs2019sv)
+#'
 #' var_case("Preventive care visits", "MAJOR", "Preventive care")
 #' tab("Preventive care visits")
+#'
 #' var_case("Surgery-related visits"
 #' , "MAJOR"
 #' , c("Pre-surgery", "Post-surgery"))
 #' tab("Surgery-related visits")
-var_case = function(newvr, vr, cases) {
+#'
+#' var_case("Non-primary"
+#' , "SPECCAT.bad"
+#' , c("Surgical care specialty", "Medical care specialty"))
+#' tab("Non-primary")
+#' tab("Non-primary", drop_na = TRUE)
+var_case = function(newvr, vr, cases, retain_na = TRUE) {
   design = .load_survey()
   nm = names(design$variables)
   assert_that(vr %in% nm, msg = paste("Variable", vr, "not in the data."))
@@ -30,6 +39,10 @@ var_case = function(newvr, vr, cases) {
   }
 
   design$variables[,newvr] = FALSE
+  if (retain_na) {
+    idx = which(is.na(design$variables[,vr]))
+    design$variables[idx, newvr] = NA
+  }
   idx = which(design$variables[,vr] %in% cases)
   design$variables[idx, newvr] = TRUE
 

docs/pkgdown.yml

@@ -4,7 +4,7 @@ pkgdown_sha: ~
 articles:
   Example-National-Ambulatory-Medical-Care-Survey-NAMCS-tables: Example-National-Ambulatory-Medical-Care-Survey-NAMCS-tables.html
   surveytable: surveytable.html
-last_built: 2024-03-29T15:28Z
+last_built: 2024-04-04T14:42Z
 urls:
   reference: https://cdcgov.github.io/surveytable/reference
   article: https://cdcgov.github.io/surveytable/articles