Як створити категоріальні змінні в r (з прикладами)

Ви можете використовувати наступний синтаксис для створення категоріальної змінної в R:

 #create categorical variable from scratch
cat_variable <- factor(c(' A ', ' B ', ' C ', ' D '))

#create categorical variable (with two possible values) from existing variable
cat_variable <- as. factor ( ifelse (existing_variable < 4, 1, 0))

#create categorical variable (with multiple possible values) from existing variable
cat_variable <- as. factor ( ifelse (existing_variable < 3, ' A ',
                          ifelse (existing_variable < 4, ' B ', 
                          ifelse (existing_variable < 5, ' C ', 
                          ifelse (existing_variable < 6, ' D ',0)))))

Наступні приклади показують, як використовувати цей синтаксис на практиці.

Приклад 1: Створення категоріальної змінної з нуля

Наступний код показує, як створити категоріальну змінну з нуля:

 #create data frame
df <- data. frame (var1=c(1, 3, 3, 4, 5),
                 var2=c(7, 7, 8, 3, 2),
                 var3=c(3, 3, 6, 10, 12),
                 var4=c(14, 16, 22, 19, 18))

#view data frame
df

var1 var2 var3 var4
1 1 7 3 14
2 3 7 3 16
3 3 8 6 22
4 4 3 10 19
5 5 2 12 18

#add categorical variable named 'type' to data frame
df$type <- factor(c(' A ', ' B ', ' B ', ' C ', ' D '))

#view updated data frame
df

var1 var2 var3 var4 type
1 1 7 3 14 A
2 3 7 3 16 B
3 3 8 6 22 B
4 4 3 10 19 C
5 5 2 12 18 D

Приклад 2: Створення категоріальної змінної (з двома значеннями) з існуючої змінної

Наступний код показує, як створити категоріальну змінну з існуючої змінної у фреймі даних:

 #create data frame
df <- data. frame (var1=c(1, 3, 3, 4, 5),
                 var2=c(7, 7, 8, 3, 2),
                 var3=c(3, 3, 6, 10, 12),
                 var4=c(14, 16, 22, 19, 18))

#view data frame
df

var1 var2 var3 var4
1 1 7 3 14
2 3 7 3 16
3 3 8 6 22
4 4 3 10 19
5 5 2 12 18

#add categorical variable named 'type' using values from 'var4' column
df$type <- as. factor ( ifelse (df$var1 < 4, 1, 0))

#view updated data frame
df

var1 var2 var3 var4 type
1 1 7 3 14 1
2 3 7 3 16 1
3 3 8 6 22 1
4 4 3 10 19 0
5 5 2 12 18 0

Використовуючи оператор ifelse() , ми створили нову категоріальну змінну під назвою “type”, яка приймає наступні значення:

• 1 , якщо значення стовпця var1 менше 4.
• 0 , якщо значення в стовпці var1 не менше 4.

Приклад 3: Створення категоріальної змінної (з кількома значеннями) з існуючої змінної

У наступному коді показано, як створити категоріальну змінну (з кількома значеннями) з існуючої змінної у кадрі даних:

 #create data frame
df <- data. frame (var1=c(1, 3, 3, 4, 5),
                 var2=c(7, 7, 8, 3, 2),
                 var3=c(3, 3, 6, 10, 12),
                 var4=c(14, 16, 22, 19, 18))

#view data frame
df

var1 var2 var3 var4
1 1 7 3 14
2 3 7 3 16
3 3 8 6 22
4 4 3 10 19
5 5 2 12 18

#add categorical variable named 'type' using values from 'var4' column
df$type <- as. factor ( ifelse (df$var1 < 3, ' A ',
                     ifelse (df$var1 < 4, ' B ', 
                     ifelse (df$var1 < 5, ' C ', 
                     ifelse (df$var1 < 6, ' D ', ' E ')))))

#view updated data frame
df

var1 var2 var3 var4 type
1 1 7 3 14 A
2 3 7 3 16 B
3 3 8 6 22 B
4 4 3 10 19 C
5 5 2 12 18 D

Використовуючи оператор ifelse() , ми створили нову категоріальну змінну під назвою “type”, яка приймає наступні значення:

• ‘ A ‘, якщо значення стовпця ‘var1’ менше 3.
• В іншому випадку « B », якщо значення в стовпці «var1» менше 4.
• В іншому випадку « C », якщо значення в стовпці «var1» менше 5.
• В іншому випадку « D », якщо значення в стовпці «var1» менше 6.
• В іншому випадку « Е ».

Додаткові ресурси

Як створити фіктивні змінні в R
Як перетворити множник на символ у R
Як перетворити символ на число в R