R/tidyup_df.R

In wadpac/GGIR: Raw Accelerometer Data Analysis

tidyup_df = function(df = c(), digits = 3) {
  
  df = df[rowSums(!is.na(df)) > 0, ]
  if (ncol(df) > 1) {
    # Round columns that can be coerced to numeric
    # But round fragmentation metrics to 6 decimal places instead of default 3
    # because some fragmentation metrics can reveal relevant info
    # at 4 or 5 decimal places
    for (fragCols in c(TRUE, FALSE)) {
      if (fragCols == TRUE) {
        colsID = grep(pattern = "FRAG_|ABI|SSP", x = colnames(df), invert = FALSE)
        digitsRound = 6
      } else {
        colsID = grep(pattern = "FRAG_|ABI|SSP", x = colnames(df), invert = TRUE)
        digitsRound = digits
      }
      myRoundFun = function(x) {
        tryCatch(round(as.numeric(as.character(x)), digits = digitsRound), 
                 error = function(cond) return(x),
                 warning = function(cond) return(x))
      }
      if (length(colsID) > 0) {
        df[, colsID] = lapply(df[, colsID], FUN = myRoundFun)
      }
    }
    # list to data frame
    df = as.data.frame(df)
  }
  return(df)
}