Export methods correctly

NAMESPACE

@@ -1,5 +1,12 @@
 # Generated by roxygen2: do not edit by hand
 
+S3method(quantiles,data.table)
+S3method(quantiles,tbl_df)
+S3method(tab,data.table)
+S3method(tab,tbl_df)
+S3method(tabcount,data.frame)
+S3method(tabcount,data.table)
+S3method(tabcount,tbl_df)
 export(quantiles)
 export(quo_to_chr)
 export(tab)

R/quantiles.R

@@ -25,3 +25,20 @@ quantiles <- function(df, ...) {
   UseMethod("quantiles", df)
   # to do: add round option like in tab()
 }
+
+#' @export
+quantiles.data.table <- function(df, ..., probs = seq(0, 1, 0.1), na.rm = FALSE) {
+  vars <- dplyr::quos(...) %>% quo_to_chr()
+  tabbed <- df[, lapply(.SD, function(x) quantile(x, probs = probs, na.rm = na.rm)),
+    .SDcols = vars
+  ][, p := probs] %>% setcolorder(c("p", vars))
+  tabbed[] # make sure it prints
+}
+
+#' @export
+quantiles.tbl_df <- function(df, ..., probs = seq(0, 1, 0.1), na.rm = FALSE) {
+  vars <- dplyr::quos(...)
+  df %>%
+    dplyr::summarize(p = list(probs), q = list(quantile(!!!vars, probs))) %>%
+    tidyr::unnest()
+}

R/tab.R

@@ -28,3 +28,34 @@
 tab <- function(df, ...) {
   UseMethod("tab", df)
 }
+
+#' @export
+tab.data.table <- function(df, ..., round=2) { # note ... is the variable names to group by
+  group_by <- dplyr::quos(...) %>% quo_to_chr()
+  rowsofdata <- df[, .N] # faster than nrow() on big data.tables
+  df[, .N, by = group_by][,
+    temp_prop := N/rowsofdata][,
+    prop := round(temp_prop, digits = round)][
+    order(-N)][, # sort in descending order by N before cumulative prop
+    cum_prop := round(cumsum(temp_prop), digits = round)][,
+    temp_prop := NULL][ # remove temp var
+    order(-N)] # make sure final data.table sorted
+}
+
+#' @export
+tab.tbl_df <- function(df, ..., round = 2) { # to check without requiring tibble
+  group_by <- dplyr::quos(...)
+  rowsofdata <- nrow(df)
+  df %>%
+    dplyr::group_by(!!!group_by) %>% # !!! since it's a quosure
+    dplyr::summarize(N = n()) %>%
+    dplyr::arrange(desc(N)) %>%
+    dplyr::ungroup() %>%
+    dplyr::mutate(
+      temp_prop = N/rowsofdata,
+      prop = round(temp_prop, digits = round),
+      cum_prop = round(cumsum(temp_prop), digits = round)
+    ) %>%
+    dplyr::select(-temp_prop)
+}
+

R/tabcount.R

@@ -27,3 +27,23 @@
 tabcount <- function(df, ...) { # note ... is the variable names to group by
   UseMethod("tabcount", df)
 }
+
+#' @export
+tabcount.data.table <- function(df, ...) {
+  group_by <- dplyr::quos(...) %>% quo_to_chr()
+  df[, .N, by = group_by][, .N]
+}
+
+#' @export
+tabcount.tbl_df <- function(df, ...) {
+  group_by <- dplyr::quos(...)
+  df %>%
+    dplyr::n_distinct(!!!group_by) # !!! since it's a quosure
+}
+
+#' @export
+tabcount.data.frame <- function(df, ...) {
+  group_by <- quo_to_chr(group_by)
+  length(unique(df[[group_by]]))
+}
+