Replace functions are modified

DESCRIPTION

@@ -20,5 +20,6 @@ LinkingTo:
     Rcpp,
     RcppArmadillo
 Suggests: 
-    testthat (>= 3.0.0)
+    testthat (>= 3.0.0),
+    pryr (>= 0.1.6) 
 Config/testthat/edition: 3

NAMESPACE

@@ -34,6 +34,8 @@ export(min_by_rownames)
 export(mkdir)
 export(mostfreq)
 export(paste_list)
+export(replace_empty_with_na)
+export(replace_na_with_zero)
 export(reverse)
 export(rotate)
 export(row_max)

R/replace.R

@@ -0,0 +1,51 @@
+#' Replace NA with zero
+#'
+#' Replace NA_integer_ or NA_real_ values with zero in a memory-efficient way
+#'
+#' @param df a data frame
+#' @return no return value
+#'
+#' @examples
+#' \donttest{df <- data.frame(x = c(1, NA, 3), y = c("A", "B", NA), z = c(NA, 5, NA))
+#' pryr::address(df)
+#' replace_na_with_zero(df)
+#' pryr::address(df)
+#' df}
+#'
+#' @export
+replace_na_with_zero <- function(df) {
+  old_class <- class(df)
+  set_dt(df)
+  class <- sapply(df, class)
+  cols <- names(class)[which(class %in% c("numeric", "integer"))]
+  df[, `:=`((cols), lapply(.SD, function(x) ifelse(is.na(x), 0, x))),
+     .SDcols = cols]
+  setattr(df, "class", old_class)
+  invisible(df)
+}
+
+#' Replace empty with NA
+#'
+#' Replace empty string like "" with NA_character_ in a memory-efficient way
+#'
+#' @param df a data frame
+#' @return no return value
+#'
+#' @examples
+#' \donttest{df <- data.frame(x = c("A", "B", ""), y = c(1, NA, 3), z = c("", "E", ""))
+#' pryr::address(df)
+#' replace_empty_with_na(df)
+#' pryr::address(df)
+#' df}
+#'
+#' @export
+replace_empty_with_na <- function(df) {
+  old_class <- class(df)
+  set_dt(df)
+  class <- sapply(df, class)
+  cols <- names(class)[which(class == "character")]
+  df[, `:=`((cols), lapply(.SD, function(x) ifelse(x == "", NA, x))),
+     .SDcols = cols]
+  setattr(df, "class", old_class)
+  invisible(df)
+}