R/ezr_get_col_index.R

In jmp1989/easyr: Helpful wrappers for common EDA, Data Manipulation, & Modeling

#' Get Column Index positions
#'
#' Get all the index positions of each column in a dataframe. You can also get a specific index value
#'
#' @param dataset  dataframe
#' @param specific Null by default.  Enter a specific column if you need it
#'
#' @return
#' @export
#'
#' @examples ezr.get_col_index(dataset=mtcars)
#'
#'
ezr.get_col_index = function(dataset, specific=NULL){

    col_positions =as.data.frame(names(dataset)) %>% mutate(position = row.names(.))
    names(col_positions ) = c('variable','position')
    col_positions = col_positions %>% dplyr::select(position, variable) #re-arranging

    if(is.null(specific)==FALSE){

        col_positions = col_positions %>% dplyr::filter(variable == specific) %>%
            dplyr::select(position) %>% as.numeric(.)

    }
    return(col_positions)

}