Fixed an issue with count treatment of non-NA elements (only presen…

…t when some elements *are* NA)
njlyon0 · May 15, 2024 · 72f8452 · 72f8452
1 parent 8ad6fbc
commit 72f8452
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/R/count.R b/R/count.R
@@ -10,7 +10,7 @@
 #' 
 #' @examples
 #' # Count instances of vector elements
-#' count(vec = c(1, 1, NA, "a", "a", 1, "a", NA, "x"))
+#' count(vec = c(1, 1, NA, "a", 1, "a", NA, "x"))
 #' 
 count <- function(vec = NULL){
 
@@ -44,8 +44,10 @@ count <- function(vec = NULL){
       # If not NA...
     } else {
 
+      no_empty_vec <- vec[!is.na(vec)]
+
       # Count occurrences
-      bit_ct <- length(vec[vec == bit])
+      bit_ct <- length(no_empty_vec[no_empty_vec == bit])
 
       # Add to dataframe
       count_df[(count_df$value == bit & !is.na(count_df$value)), ]$count <- bit_ct

diff --git a/man/count.Rd b/man/count.Rd