R/trim_quantiles.R

In andrie/discretizer: Discretizes values

#' Trim x and y range based on quantiles
#'
#' @param x A data frame with columns x and y
#' @param q Quantiles cutoff
#'
#' @return A data frame
#' @export
#'
#' @importFrom stats na.omit quantile
#'
trim_quantiles <- function(x, q = 0.05){
  x_range <- quantile(x$x, probs = c(q, 1-q), na.rm = TRUE)
  y_range <- quantile(x$y, probs = c(q, 1-q), na.rm = TRUE)

  x$x[x$x < x_range[1]] <- NA
  x$x[x$x > x_range[2]] <- NA
  x$y[x$y < y_range[1]] <- NA
  x$y[x$y > x_range[2]] <- NA
  # browser()
  na.omit(x)
}