R/data_split.R

In goutham1220/shipshape: Intuitive Calculation and Analysis of Procrustes & Elastic Distance Metrics

#' split a matrix into desired 3-D array at set interval.
#'
#'@param mat input matrix
#'@param interval interval for splitting of matrix
#'
#' @return 3-D array
#'
#' @author Goutham Swaminathan \email{goutham1220@gmail.com}
#'
#' @export
#'
#' @examples
#'
#' \dontrun{
#' test_data = bone_list$x
#' test_data = cbind(test_data, bone_list$y)
#' bones = data_split(test_data, 100)
#' }
#'

data_split <- function(mat, interval){

  z = nrow(mat)/interval

  arr = array(dim = c(interval, ncol(mat), z))

  arr[,,1] = mat[1:interval,]

  for(i in 2:z){

    start = (i-1) * interval + 1
    end = i * interval
    arr[,,i] = mat[start:end,]

  }

  return(arr)

}