Как сделать одну переменную фактора горячего кодирования, которая имеет более 3 уровней?

Я хочу представить факторные переменные как значения 0 и 1 с помощью одного горячего кодирования в r как data.frame.

Среди факторных переменных я хотел бы выполнить одно горячее кодирование только для переменных с тремя или более уровнями.

Это мой код R.

german<-read.csv("http://freakonometrics.free.fr/german_credit.csv", header=TRUE)
F=c(1,2,4,5,7,8,9,10,11,12,13,15,16,17,18,19,20,21)
for(i in F) german[,i]=as.factor(german[,i])
str(german)
'data.frame':   1000 obs. of  21 variables:
 $ Creditability                    : Factor w/ 2 levels "0","1": 2 2 2 2 2 2 2 2 2 2 ...
 $ Account.Balance                  : Factor w/ 4 levels "1","2","3","4": 1 1 2 1 1 1 1 1 4 2 ...
 $ Duration.of.Credit..month.       : int  18 9 12 12 12 10 8 6 18 24 ...
 $ Payment.Status.of.Previous.Credit: Factor w/ 5 levels "0","1","2","3",..: 5 5 3 5 5 5 5 5 5 3 ...
 $ Purpose                          : Factor w/ 10 levels "0","1","2","3",..: 3 1 9 1 1 1 1 1 4 4 ...
 $ Credit.Amount                    : int  1049 2799 841 2122 2171 2241 3398 1361 1098 3758 ...
 $ Value.Savings.Stocks             : Factor w/ 5 levels "1","2","3","4",..: 1 1 2 1 1 1 1 1 1 3 ...
 $ Length.of.current.employment     : Factor w/ 5 levels "1","2","3","4",..: 2 3 4 3 3 2 4 2 1 1 ...
 $ Instalment.per.cent              : Factor w/ 4 levels "1","2","3","4": 4 2 2 3 4 1 1 2 4 1 ...
 $ Sex...Marital.Status             : Factor w/ 4 levels "1","2","3","4": 2 3 2 3 3 3 3 3 2 2 ...
 $ Guarantors                       : Factor w/ 3 levels "1","2","3": 1 1 1 1 1 1 1 1 1 1 ...
 $ Duration.in.Current.address      : Factor w/ 4 levels "1","2","3","4": 4 2 4 2 4 3 4 4 4 4 ...
 $ Most.valuable.available.asset    : Factor w/ 4 levels "1","2","3","4": 2 1 1 1 2 1 1 1 3 4 ...
 $ Age..years.                      : int  21 36 23 39 38 48 39 40 65 23 ...
 $ Concurrent.Credits               : Factor w/ 3 levels "1","2","3": 3 3 3 3 1 3 3 3 3 3 ...
 $ Type.of.apartment                : Factor w/ 3 levels "1","2","3": 1 1 1 1 2 1 2 2 2 1 ...
 $ No.of.Credits.at.this.Bank       : Factor w/ 4 levels "1","2","3","4": 1 2 1 2 2 2 2 1 2 1 ...
 $ Occupation                       : Factor w/ 4 levels "1","2","3","4": 3 3 2 2 2 2 2 2 1 1 ...
 $ No.of.dependents                 : Factor w/ 2 levels "1","2": 1 2 1 2 1 2 1 2 1 1 ...
 $ Telephone                        : Factor w/ 2 levels "1","2": 1 1 1 1 1 1 1 1 1 1 ...
 $ Foreign.Worker                   : Factor w/ 2 levels "1","2": 1 1 1 2 2 2 2 2 1 1 ...

Здесь я хочу горячо кодировать факторные переменные, которые имеют более 3 уровней.

Например, переменная Guarantors имеет 3 уровня 1,2,3. В результате я хочу получить переменные Guarantors1, Guarantors2 и Guarantors3, которые имеют значение только 0,1 в качестве data.frame.


r one-hot-encoding dummy-variable
person 신익수    schedule 27.08.2017    source источник
comment
Покажите нам, что вы пробовали самостоятельно. Мы не занимаемся написанием кода.   -  person emilliman5    schedule 27.08.2017
comment
Возможный дубликат Создание новых столбцов фиктивных переменных из категориальной переменной   -  person Aramis7d    schedule 28.08.2017

Ответы (1)


arrow_upward
2
arrow_downward

dplyr & purrr подход

library(dplyr)
library(purrr)

german<-read.csv("http://freakonometrics.free.fr/german_credit.csv", header=TRUE)

cols <- c(1,2,4,5,7,8,9,10,11,12,13,15,16,17,18,19,20,21)

map_df(german[, cols], as.factor) %>% 
      select_if(function(x) nlevels(x) >= 2) %>% 
      model.matrix(~. -1, data = .) %>% 
      as.data.frame()

Я бы рекомендовал прочитать справку model.matrix или другое вопросы от SO по этой теме.

person Jake Kaupp    schedule 27.08.2017