Преобразуйте один столбец из категориального в двоичный, оставьте остальные
У меня есть средний большой фрейм данных, для которого я хочу преобразовать один столбец с категориями в двоичные столбцы, по одному для каждой категории.
В то же время я хочу сохранить остальные столбцы в кадре данных.
Какой самый простой способ добиться этого?
Вот пример того, что я хочу сделать:
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