Define data frame method for dw_data_xtabulates object?

NAMESPACE

@@ -1,5 +1,7 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(as.data.frame,dw_data_tabulates)
+S3method(as.data.frame,dw_data_xtabulate)
 S3method(as.double,parameters_kurtosis)
 S3method(as.double,parameters_skewness)
 S3method(as.double,parameters_smoothness)

R/data_tabulate.R

@@ -25,6 +25,7 @@
 #' percentages to be calculated. Only applies to crosstables, i.e. when `by` is
 #' not `NULL`. Can be `"row"` (row percentages), `"column"` (column percentages)
 #' or `"full"` (to calculate relative frequencies for the full table).
+#' @param add_total Add total.
 #' @param ... not used.
 #' @inheritParams extract_column_names
 #'
@@ -380,6 +381,37 @@ insight::print_html
 insight::print_md
 
 
+#' @rdname data_tabulate
+#' @export
+as.data.frame.dw_data_tabulates <- function(x, add_total = FALSE, ...) {
+  # extract variables of frequencies
+  selected_vars <- lapply(x, function(i) attributes(i)$varname)
+  # coerce to data frame, remove rownames
+  data_frames <- lapply(x, function(i) {
+    if (add_total) {
+      out <- as.data.frame(format(i))
+      for (i in 2:ncol(out)) {
+        out[[i]] <- as.numeric(out[[i]])
+      }
+      out <- remove_empty_rows(out)
+    } else {
+      out <- as.data.frame(i)
+    }
+    rownames(out) <- NULL
+    out
+  })
+  # create nested data frame
+  data.frame(
+    var = selected_vars,
+    table = I(data_frames),
+    stringsAsFactors = FALSE
+  )
+}
+
+#' @export
+as.data.frame.dw_data_xtabulate <- as.data.frame.dw_data_tabulates
+
+
 #' @export
 format.dw_data_tabulate <- function(x, format = "text", big_mark = NULL, ...) {
   # convert to character manually, else, for large numbers,

R/data_xtabulate.R

@@ -74,6 +74,7 @@
   attr(out, "total_n") <- total_n
   attr(out, "weights") <- weights
   attr(out, "proportions") <- proportions
+  attr(out, "varname") <- obj_name
 
   class(out) <- c("dw_data_xtabulate", "data.frame")