avoid removing data if nothing matches cycles filter

addisonklinke · Apr 17, 2020 · c3c1b11 · c3c1b11
1 parent 9fc4017
commit c3c1b11
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/R/CountCycles.numeric.R b/R/CountCycles.numeric.R
@@ -98,7 +98,10 @@ CountCycles.numeric <- function(x) {
   # Format and return
   cycles <- as.data.frame(cycles)
   cycles <- plyr::rename(cycles, c("V1" = "amplitude", "V2" = "mean", "V3" = "cycles"))
-  cycles <- cycles[-which(cycles$cycles == 0), ]
+  remove = which(cycles$cycles == 0)
+  if (length(remove) > 0) {
+    cycles <- cycles[-remove, ]
+  }
   class(cycles) <- c("rainflow", "data.frame")
   return(cycles)
 }