Reescribir como un bucle para establecer variables seleccionadas en marcos de datos similares como factores, en R

Nov 24 2020

Tengo muchos años de datos, cada uno en un marco de datos idéntico, quiero poner todos los años en una lista y luego escribir uno para bucle en lugar de repetir el comando para cada año.

#set dummies as factors (these dummies repeat across years)
mydummies<-c ('hru_i', 'ge_nonngsother_i','ge_sgt_i')
DF2012[,mydummies]<-lapply(DF2012[,mydummies],factor)
DF2013[,mydummies]<-lapply(DF2013[,mydummies],factor)

Intenté poner todos los marcos de datos en una lista para poder ejecutar un bucle, pero no cambió los marcos de datos ...

df.list<- list(DF2012,DF2013)
#want to create a loop here

Respuestas

1 GregorThomas Nov 24 2020 at 20:12

Un anidado lapplyfuncionará para esto:

df.list <- lapply(df.list, function(d) {
  d[mydummies] <- lapply(d[mydummies], factor)
  d
})

Aquí hay un ejemplo reproducible que demuestra que funciona:

df.list = list(a = head(mtcars), b = head(mtcars))
mydummies = c("cyl", "am")
sapply(df.list, sapply, class)
#      a         b        
# mpg  "numeric" "numeric"
# cyl  "numeric" "numeric"
# disp "numeric" "numeric"
# hp   "numeric" "numeric"
# drat "numeric" "numeric"
# wt   "numeric" "numeric"
# qsec "numeric" "numeric"
# vs   "numeric" "numeric"
# am   "numeric" "numeric"
# gear "numeric" "numeric"
# carb "numeric" "numeric"

df.list <- lapply(df.list, function(d) {
  d[mydummies] <- lapply(d[mydummies], factor)
  d
})

#      a         b        
# mpg  "numeric" "numeric"
# cyl  "factor"  "factor" 
# disp "numeric" "numeric"
# hp   "numeric" "numeric"
# drat "numeric" "numeric"
# wt   "numeric" "numeric"
# qsec "numeric" "numeric"
# vs   "numeric" "numeric"
# am   "factor"  "factor" 
# gear "numeric" "numeric"
# carb "numeric" "numeric"
1 akrun Nov 24 2020 at 20:24

Nosotros podemos usar tidyverse

library(dplyr)
library(purrr)
df.list <- map(df.list, ~ .x %>% 
                    mutate(across(all_of(mydummies), factor)))

datos

df.list <- list(a = head(mtcars), b = head(mtcars))
mydummies <- c("cyl", "am")