R/reshape_data.R

In leosanchezsoler/r-project-the-bridge: GapminderApp

#' @title reshape_data
#' 
#' @description This function is called by skeleton and reshapes the df to a [n, 2] shape
#' 
#' @param data list of datasets
#' @param config List of configuration parameters 
#'
#' @return
#' 
#' @import logging
#' 
reshape_data <- function(config, lista_pred_target){
  
  
  for (i in 1:length(lista_pred_target$predictoras)){
    
    col = colnames(lista_pred_target$predictoras[[i]])
    lista_pred_target$predictoras[[i]] <- reshape2::melt(data = lista_pred_target$predictoras[[i]], 
                                                   id.vars = 'country', 
                                                   measure.vars = col[!(col %in% "country")],
                                                   value.name = config$data$predictors[i]
                                                   )
  }
  
  col = colnames(lista_pred_target$target)
  lista_pred_target$target <- reshape2::melt(data = lista_pred_target$target,
                                       id.vars = 'country',
                                       measure.vars = col[!(col %in% "country")],
                                       value.name = config$data$target
                                       )
  
  
  
  df <- lista_pred_target$predictoras[[1]]
  
  for (i in 2:length(lista_pred_target$predictoras)){
    
    df <- merge(df, lista_pred_target$predictoras[[i]], by = c('country', 'variable'))
  }
  
  df <- merge(df, lista_pred_target$target, by = c('country', 'variable'))
  
  predict_Y <- df[(df$variable == config$data$prediction$year & df$country == config$data$prediction$country), ]
  
  datos <- na.omit(df)
  
  return(list(datos, predict_Y))
  
}