Przepisz jako pętlę, aby ustawić wybrane zmienne w podobnych ramkach danych jako czynniki, w R

Question

Mam dane z wielu lat, każdy w identycznej ramce danych, chcę umieścić wszystkie lata na liście, a następnie napisać pętlę for zamiast powtarzać polecenie dla każdego roku.

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

Próbowałem umieścić wszystkie ramki danych na liście, aby móc uruchomić pętlę, ale nie zmieniło to ramek danych ...

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

Answer 1

Zagnieżdżone lapplybędzie działać w tym przypadku:

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

Oto odtwarzalny przykład pokazujący, że to działa:

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"

Answer 2

Możemy użyć tidyverse

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

dane

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