2015-10-16 22 views
31

Mam rama przykładowe dane jak poniżej:przymusić wiele kolumn do czynników naraz

data <- data.frame(matrix(sample(1:40), 4, 10, dimnames = list(1:4, LETTERS[1:10]))) 

Chcę wiedzieć, w jaki sposób można zaznaczyć kilka kolumn i konwertować je ze sobą czynników. Zwykle robię to w taki sposób, jak data$A = as.factor(data$A). Ale gdy ramka danych jest bardzo duża i zawiera wiele kolumn, ten sposób będzie bardzo czasochłonny. Czy ktoś wie, czy istnieje jakiś sprytny sposób to zrobić?

Odpowiedz

48

wybrać kilka kolumn, aby zmusić do czynników:

cols <- c("A", "C", "D", "H") 

Zastosowanie lapply() zniewalać i zastąpić wybrane kolumny:

data[cols] <- lapply(data[cols], factor) 

Sprawdź rezultat:

sapply(data, class) 
#  A   B   C   D   E   F   G 
# "factor" "integer" "factor" "factor" "integer" "integer" "integer" 
#  H   I   J 
# "factor" "integer" "integer" 
+0

Czy nie musi to być 'data [, cols] <- lapply (data [, cols], factor)' (z głównym przecinkiem dla kolumn)? – Tgsmith61591

+4

@ Tgsmith61591- To może być albo. Przecinek jest podzbiorem typu macierzowego, a przecinek nie jest podzbiorem listy. Ramki danych mogą być podsegmentowane przez jedną z nich, więc każda z nich zadziała. –

21

Oto opcja przy użyciu dplyr. Operator %<>% z magrittr zaktualizował obiekt LHS z wartością wynikową.

library(magrittr) 
library(dplyr) 
cols <- c("A", "C", "D", "H") 

data %<>% 
     mutate_each_(funs(factor(.)),cols) 
str(data) 
#'data.frame': 4 obs. of 10 variables: 
# $ A: Factor w/ 4 levels "23","24","26",..: 1 2 3 4 
# $ B: int 15 13 39 16 
# $ C: Factor w/ 4 levels "3","5","18","37": 2 1 3 4 
# $ D: Factor w/ 4 levels "2","6","28","38": 3 1 4 2 
# $ E: int 14 4 22 20 
# $ F: int 7 19 36 27 
# $ G: int 35 40 21 10 
# $ H: Factor w/ 4 levels "11","29","32",..: 1 4 3 2 
# $ I: int 17 1 9 25 
# $ J: int 12 30 8 33 

Lub jeśli używamy data.table, albo użyć for pętlę z set

setDT(data) 
for(j in cols){ 
    set(data, i=NULL, j=j, value=factor(data[[j]])) 
} 

Albo możemy określić 'cols' w .SDcols i przypisać (:=) RHS " cols '

setDT(data)[, (cols):= lapply(.SD, factor), .SDcols=cols] 
7

nowsza tidyverse sposobem jest użycie mutate_at funkcję:

library(tidyverse) 
library(magrittr) 
set.seed(88) 

data <- data.frame(matrix(sample(1:40), 4, 10, dimnames = list(1:4, LETTERS[1:10]))) 
cols <- c("A", "C", "D", "H") 

data %<>% mutate_at(cols, funs(factor(.))) 
str(data) 
$ A: Factor w/ 4 levels "5","17","18",..: 2 1 4 3 
$ B: int 36 35 2 26 
$ C: Factor w/ 4 levels "22","31","32",..: 1 2 4 3 
$ D: Factor w/ 4 levels "1","9","16","39": 3 4 1 2 
$ E: int 3 14 30 38 
$ F: int 27 15 28 37 
$ G: int 19 11 6 21 
$ H: Factor w/ 4 levels "7","12","20",..: 1 3 4 2 
$ I: int 23 24 13 8 
$ J: int 10 25 4 33 
3

, a dla kompletności i w odniesieniu do this question asking about changing string columns only, tam mutate_if:

data <- cbind(stringVar = sample(c("foo","bar"),10,replace=TRUE), 
       data.frame(matrix(sample(1:40), 10, 10, dimnames = list(1:10, LETTERS[1:10]))),stringsAsFactors=FALSE)  

factoredData = data %>% mutate_if(is.character,funs(factor(.))) 
0

Jeśli masz inny cel uzyskanie w wartościach od tabelę można następnie przekonwertować na następującą metodę: można wypróbować w następujący sposób:

### pre processing 
ind <- bigm.train[,lapply(.SD,is.character)] 
ind <- names(ind[,.SD[T]]) 
### Convert multiple columns to factor 
bigm.train[,(ind):=lapply(.SD,factor),.SDcols=ind] 

Ten egzemplarz ect kolumny, które są oparte na znakach, a następnie konwertuje je na czynnik.