Schreiben Sie als Schleife neu, um ausgewählte Variablen in ähnlichen Datenrahmen als Faktoren in R festzulegen

Nov 24 2020

Ich habe viele Jahre Daten, jede in einem identischen Datenrahmen. Ich möchte alle Jahre in eine Liste einfügen und dann eine for-Schleife schreiben, anstatt den Befehl für jedes Jahr zu wiederholen.

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

Ich habe versucht, alle Datenrahmen in eine Liste aufzunehmen, damit ich eine Schleife ausführen kann, aber die Datenrahmen wurden dadurch nicht geändert ...

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

Antworten

1 GregorThomas Nov 24 2020 at 20:12

Ein verschachteltes lapplywird dafür funktionieren:

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

Hier ist ein reproduzierbares Beispiel, das zeigt, dass es funktioniert:

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

Wir können benutzen tidyverse

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

Daten

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