R/CreateLVIndices.R

In oharar/LatentINLA: Package to Write GLLVMs in INLA

#' Format a data frame/matrix so that it is in the right format for INLA
#'
#' @param dat data. A data frame or matrix
#' @param nLVs The number of latent variables required
#' @return A data frame with the same number of columns, but each row only has one value in it (the rest are NAs)
#' @examples
#' CreateLVIndices(matrix(1:10, ncol=5), nLVs=1)
#' @export


CreateLVIndices <- function(dat, nLVs=1) {
  if(!is.data.frame(dat) & !is.matrix(dat)) stop("dat should be a matrix or data frame")
  if(!is.data.frame(dat) & !is.matrix(dat)) stop("dat should be a matrix or data frame")
  res <- sapply(nLVs:ncol(dat), MakeIDs, df=dat)
  colnames(res) <- c("L", paste0("col", (1+nLVs):ncol(dat)))
  data.frame(res)
}