R/get_pca_res.R

In thierrycnam/nbafuns: Functions for nba PCA analysis

#' get_pca_res
#'
#' @param df_1 df
#' @param quanti_sup suppl. quantitative variable
#' @param quali_sup suppl. qualitative variable
#'
#' @return data frame
#' @export
#' @import FactoMineR
#' @import factoextra
#' @import tidyr
#' @import dplyr
#'
#' @examples
#' df_1 <- nba_stats
#' quanti_sup <- c("rk", "age", "g", "mp")
#' quali_sup <- c("player", "pos", "tm")
#' get_pca_res(df_1, quanti_sup, quali_sup)
get_pca_res <- function(df_1, quanti_sup, quali_sup) {
  quali_sup_indexes <- sapply(quali_sup,
                              function(x) {which(colnames(df_1)== x)})
  quanti_sup_indexes <- sapply(quanti_sup,
                               function(x) {which(colnames(df_1)== x)})
  df_1 <-
    df_1 %>%
    drop_na()
  pca_res <- PCA(df_1, scale.unit = TRUE,
                 quanti.sup = quanti_sup_indexes,
                 quali.sup = quali_sup_indexes,
                 graph = FALSE)
}