Как преобразовать значения столбца фрейма данных в 0 и 1?

Question

Я работаю над моделью отрицательной биномиальной регрессии, которая прогнозирует количество инициированных членских счетов частных членов парламентами в Японии на основе их доли голосов, возраста, пола и должности в парламенте. Чтобы вычислить AME для переменной parliamentary_office, мне нужно создать два новых фрейма данных df.0 и df.1. В качестве примера, вот фрейм данных df.0:

  (Intercept) voteshare age sexmale term parliamentary_office
1           1     37.92  57    male    0                    0
2           1     45.99  65    male    5                    0
3           1     36.18  59  female    3                    0
4           1      43.3  47    male    1                    0
5           1     45.48  58    male    5                    0
6           1     31.89  44    male    0                    0

Как преобразовать столбец sexmale в числа?

Вот мой код:

#rm(list=ls())
library(foreign)
dat <- read.dta(file = 'activity.dta', convert.factors = FALSE)
dat_clear <- na.omit(dat)
datc_2012 <- dat_clear[dat_clear$election == 2012, ]

library(MASS)
summary(m2.negbin <- glm.nb(num_pmbs_initiated ~
voteshare + age + sex + term
+ parliamentary_office, data = datc_2012,
link = "log"))

df.0 <- data.frame(cbind(1,
m2.negbin[["model"]]$voteshare,
m2.negbin[["model"]]$age,
m2.negbin[["model"]]$sex,
m2.negbin[["model"]]$term,
m2.negbin[["model"]]$parliamentary_office))

colnames(df.0) <- names(coef(m2.negbin))
df.1 <- df.0
df.0[,"parliamentary_office"] <- 0
df.1[,"parliamentary_office"] <- 1 

Answer 1

вы можете использовать функцию ifelse, если у вас есть только мужчины и женщины, и вам нужно изменить столбец на числа 0-1.

df.0$sexmale <- sapply(df.0$sexmale, function (x) ifelse(x == "male", 0, 1))