R/set_window.R

In badbayesian/twitchNLP: Toolkit to do NLP analysis of twitch comments streams.

#' Sets moving window average
#' 
#' @param df
#' @param window_size 
#' @param .f
#' @return
set_window <- function(df, window_size, .f = sum) {
  window_size  <- window_size  / 2
  df_ <- df %>%
    dplyr::count(elapsed_time) %>%
    dplyr::mutate(count =
                    slider::slide_index(.x = n,
                                        .i = elapsed_time,
                                        .f = .f,
                                        .before = 60 * window_size,
                                        .after = 60 * window_size) %>%
                    unlist(),
                  count = 2 * count / window_size )
  dplyr::left_join(df, df_, by = "elapsed_time")
}