Преобразуйте один столбец из категориального в двоичный, оставьте остальные

У меня есть средний большой фрейм данных, для которого я хочу преобразовать один столбец с категориями в двоичные столбцы, по одному для каждой категории.

В то же время я хочу сохранить остальные столбцы в кадре данных.

Какой самый простой способ добиться этого?

Вот пример того, что я хочу сделать:

d<-data.frame(ID=c("a","b","c","d"), Gender=c("male", "male", "female","female"), Age =c(23,45,18,11))

 ID Gender Age
1  a   male  23
2  b   male  45
3  c female  18
4  d female  11

впоследствии должен выглядеть как d2, так что столбцы ID и Age все еще там и не тронуты:

d2<-data.frame(ID=c("a","b","c","d"), Gender.male=c(1, 1, 0, 0), Gender.female=c(0,0,1,1), Age =c(23,45,18,11))

  ID Gender.male Gender.female Age
1  a           1             0  23
2  b           1             0  45
3  c           0             1  18
4  d           0             1  11

3 ответа

Решение

Мы можем использовать spread

library(tidyvesre)
d %>% 
  mutate(n = 1) %>% 
  spread(Gender, n, fill = 0)

Или использовать dcast от reshape2

library(reshape2)
dcast(d, ID + Age ~ Gender, length)
#  ID Age female male
#1  a  23      0    1
#2  b  45      0    1
#3  c  18      1    0
#4  d  11      1    0

Мы можем использовать dummies пакет.

library(dummies)

d2 <- dummy("Gender", d)
d3 <- cbind(d, d2)
d3$Gender <- NULL
d3
#   ID Age Genderfemale Gendermale
# 1  a  23            0          1
# 2  b  45            0          1
# 3  c  18            1          0
# 4  d  11            1          0
Другие вопросы по тегам