R/md.pattern.R

In yikeshu0611/nhanesR: Tools for "NHANES" Database

#' missing pattern
#'
#' @param data dataframe or list
#'
#' @return missing pattern for each year and overall
#' @export
#'
md.pattern <- function(data){
    if (is.data.frame(data)){
        for (i in unique(data$Year)) {
            cat(crayon::red(i),'\n')
            x <- mice::md.pattern(plot = FALSE,
                                  data[data$Year==i,set::not(colnames(data),'Year','seqn','sdmvpsu','sdmvstra')])
            print(t(x))
        }
        cat(crayon::red('Overall\n'))
        x <- mice::md.pattern(plot = FALSE,data[,set::not(colnames(data),'Year','seqn','sdmvpsu','sdmvstra')])
        print(t(x))
    }else{
        for (i in 1:length(data)) {
            cat(crayon::red(names(data)[i]),'\n')
            x <- mice::md.pattern(plot = FALSE,
                                  data[[i]][,set::not(colnames(data[[i]]),'Year','seqn','sdmvpsu','sdmvstra')])
            print(t(x))
        }
        cat(crayon::red('Overall\n'))
        data <- do.call(plyr::rbind.fill,data)
        x <- mice::md.pattern(plot = FALSE,data[,set::not(colnames(data),'Year','seqn','sdmvpsu','sdmvstra')])
        print(t(x))
    }

}