如何在R中重新编码变量

Question

我正在尝试在R数据框中重新编码变量。 示例-我数据集中的变量X包含1和0。 我想创建另一个变量Y，分别将X的1和0分别编码为Yes和No。

我试图这样做来创建重新编码的Y变量：

w <- as.character()

for (i in seq_along(x))  {
    if (x[i] == 1)  {
        recode <- "Yes"
    } else if (x[i] == 0)  {
        recode <- "No"       
    }
    w <- cbind(w, recode)
}

然后，我这样做是为了将X和Y排列在一起：

y <- c(x, y)

我回来的是：

 y
 # [1] "1"   "1"   "0"   "1"   "0"   "0"   "1"   "1"   "0"   "1"   "0"   "0"   "Yes" "Yes" "No"  "Yes" "No"  "No" 

我期待带有X和Y列的数据框。

题：

1. 如何将X和Y放入数据框？
2. 有没有更好的方法来重新编码数据帧中的变量？

Answer 1

重新编码通常是将新标签应用于因子（分类变量）的级别

在R中，您可以这样操作：

w <- factor(x, levels = c(1,0), labels = c('yes', 'no'))

Answer 2

使用以下数据：

x  <- c(rep.int(0, 10), rep.int(1, 10))
df <- as.data.frame(x)
df
#    x
# 1  0
# 2  0
# 3  0
# ...

我将创建一个新变量并一步重新编码：

df$y[df$x == 1] <- "yes"
df$y[df$x == 0] <- "no"
df
#    x   y
# 1  0  no
# 2  0  no
# 3  0  no
# ...
# 11 1 yes
# 12 1 yes
# 13 1 yes
# ...

请注意for循环在R中并非最佳，但您的循环基本上是正确的。 您需要在循环本身w <- cbind(w, recode)替换为w <- rbind(w, recode) w <- cbind(w, recode) ，最后一步，您可以cbind x和w cbind ：

w <- as.character()
for (i in seq_along(x))  {
  if (x[i] == 1)  {
    recode <- "Yes"
  } else if (x[i] == 0)  {
    recode <- "No"       
  }
  w <- rbind(w, recode)
}
y <- c(x, w)
y

rbind()追加行， cbind()追加列， c()将两个字符串连接在一起，这就是为什么要将两个列表连接在一起的原因。

Answer 3

这是您实际上不应在R中使用循环的众多情况之一。

而是使用向量化，即ifelse或索引。

result = data.frame(x = x, y = ifelse(x == 1, 'yes', 'no'))

（这假定输入中只有1和0；如果不是这种情况，则需要嵌套的ifelse或包含翻译的列表）。