1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
|
#' Count entries in a factor
#'
#' @param f A factor (or character vector).
#' @param sort If `TRUE`, sort the result so that the most common values
#' float to the top.
#' @param prop If `TRUE`, compute the fraction of marginal table.
#' @return A tibble with columns `f`, `n` and `p`, if prop is `TRUE`.
#' @export
#' @examples
#' f <- factor(sample(letters)[rpois(1000, 10)])
#' table(f)
#' fct_count(f)
#' fct_count(f, sort = TRUE)
#' fct_count(f, sort = TRUE, prop = TRUE)
fct_count <- function(f, sort = FALSE, prop = FALSE) {
f <- check_factor(f)
check_bool(sort)
check_bool(prop)
n_na <- sum(is.na(f))
n <- c(tabulate(f, nlevels(f)), if (n_na > 0) n_na)
df <- tibble::tibble(
f = fct_unique(f),
n = n
)
if (sort) {
df <- df[order(df$n, decreasing = TRUE), ]
}
if (prop) {
df$p <- prop.table(df$n)
}
df
}
|
