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")